hyrax.models.hsc_dcae

Classes

ArcsinhActivation	Helper module for HSCDAE to use the arcsinh function
HSCDCAE	This autoencoder is designed to work with datasets that are prepared with Hyrax's HSC Data Set class.

Module Contents

class ArcsinhActivation(*args, **kwargs)

Bases: torch.nn.Module

Helper module for HSCDAE to use the arcsinh function

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

forward(x)

class HSCDCAE(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.

encoder1

encoder2

encoder3

encoder4

pool

decoder4

decoder3

decoder2

decoder1

activation

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