hyrax.models.hyrax_loopback# Attributes# logger 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_batch(batch)[source]# Training is a noop validate_batch(batch)[source]# Validation is just a forward pass test_batch(batch)[source]# Testing is just a forward pass infer_batch(batch)[source]# Inference is just a forward pass