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
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
-