hyrax.models.hyrax_loopback
===========================

.. py:module:: hyrax.models.hyrax_loopback


Attributes
----------

.. autoapisummary::

   hyrax.models.hyrax_loopback.logger


Classes
-------

.. autoapisummary::

   hyrax.models.hyrax_loopback.HyraxLoopback


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

.. py:data:: logger

.. py:class:: HyraxLoopback(config, data_sample=None)

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


   Simple model for testing which returns its own input

   Initialize internal Module state, shared by both nn.Module and ScriptModule.


   .. py:attribute:: unused_module


   .. py:attribute:: config


   .. py:attribute:: load


   .. py:method:: forward(x)

      We simply return our input



   .. py:method:: train_step(batch)

      Training is a noop