src.gridmind.feature_construction.cnn_feature_extractor
=======================================================

.. py:module:: src.gridmind.feature_construction.cnn_feature_extractor


Classes
-------

.. autoapisummary::

   src.gridmind.feature_construction.cnn_feature_extractor.ResNetFeatureExtractor


Module Contents
---------------

.. py:class:: ResNetFeatureExtractor(output_dim=512)

   Bases: :py:obj:`torch.nn.Module`


   .. py:attribute:: features


   .. py:attribute:: pool


   .. py:attribute:: flatten


   .. py:attribute:: output_dim
      :value: 512



   .. py:method:: forward(x)