hyrax.models.hyrax_loopback

Attributes

logger

Classes

HyraxLoopback

Simple model for testing which returns its own input

Module Contents

logger

class HyraxLoopback(config, data_sample=None)

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

config

load

forward(x)

We simply return our input

train_step(batch)

Training is a noop