unet_interpolated

Modification in U-Net model for low resolution gfas-frp data which requires interpolation due to non-whole number scaling required in the final layer.

Module Contents

class unet_interpolated.Model(hparams)

Bases: model.unet.Model

Inheritance diagram of unet_interpolated.Model

The primary module containing all the training functionality. It is equivalent to PyTorch nn.Module in all aspects.

forward(self, x)

Forward pass

test_epoch_end(self, outputs)

Called at the end of testing epoch to aggregate outputs.

Parameters

outputs (list) – List of individual outputs of each testing step.

Returns

Loss and logs.

Return type

dict