hyrax.models.hyrax_loopback#

Attributes#

Classes#

HyraxLoopback

Simple model for testing which returns its own input

Module Contents#

logger[source]#
class HyraxLoopback(config, data_sample=None)[source]#

Bases: torch.nn.Module

Simple model for testing which returns its own input

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

unused_module[source]#
config[source]#
load[source]#
forward(x)[source]#

We simply return our input

train_step(batch)[source]#

Training is a noop