Models
==============

Here we provide the code for the models and the layers used. Everything has been implemented in PyTorch.

.. automodule:: ccsd.src.models.ScoreNetwork_A_CC
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: ccsd.src.models.ScoreNetwork_F
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: ccsd.src.models.ScoreNetwork_A
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: ccsd.src.models.ScoreNetwork_X
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: ccsd.src.models.hodge_attention
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: ccsd.src.models.hodge_layers
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: ccsd.src.models.attention
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: ccsd.src.models.layers
   :members:
   :undoc-members:
   :show-inheritance:
