ISubGVQA.models.scene_graph_encoder
===================================

.. py:module:: ISubGVQA.models.scene_graph_encoder


Classes
-------

.. autoapisummary::

   ISubGVQA.models.scene_graph_encoder.SceneGraphEncoder


Functions
---------

.. autoapisummary::

   ISubGVQA.models.scene_graph_encoder.get_gt_scene_graph_encoding_layer


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

.. py:class:: SceneGraphEncoder(hidden_dim, dist=False)

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


   .. py:attribute:: scene_graphs_helper


   .. py:attribute:: sg_vocab


   .. py:attribute:: hidden_dim


   .. py:attribute:: dist
      :value: False



   .. py:attribute:: sg_emb_dim
      :value: 300



   .. py:attribute:: sg_vocab_embedding


   .. py:attribute:: scene_graph_encoding_layer


   .. py:attribute:: graph_layer_norm


   .. py:attribute:: bbox_encoding


   .. py:attribute:: feat_reduc


   .. py:method:: forward(x, edge_index, edge_attr, batch, explainer=False, explainer_stage=False, gt_scene_graphs=None)


.. py:function:: get_gt_scene_graph_encoding_layer(num_node_features, num_edge_features, hidden_dim)