hyrax.models.hsc_autoencoder

Classes

HSCAutoencoder

This autoencoder is designed to work with datasets that are prepared with Hyrax's HSC Data Set class.

Module Contents

class HSCAutoencoder(config, data_sample=None)

Bases: torch.nn.Module

This autoencoder is designed to work with datasets that are prepared with Hyrax’s HSC Data Set class.

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

encoder

decoder

config

forward(x)

train_step(batch)

This function contains the logic for a single training step. i.e. the contents of the inner loop of a ML training process.

Parameters:

batch (tuple) – A tuple containing the two values the loss function

Returns:

Current loss value – Dictionary containing the loss value for the current batch.

Return type:

dict