unet_downsampled

The original U-Net model with a downsampling layer at the end to match with the FWI-Reanalysis resolution.

Module Contents

class unet_downsampled.Model(hparams)

Bases: model.unet.Model

This class implements modified U-Net module by downsampling the output to match with the resolution of fwi-reanalysis. It is equivalent to PyTorch’s nn.Module in all aspects.

param LightningModule

The Pytorch-Lightning module derived from nn.module with

useful hooks

type LightningModule

nn.Module