:mod:`unet_downsampled`
=======================

.. py:module:: unet_downsampled

.. autoapi-nested-parse::

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



Module Contents
---------------

.. py:class:: Model(hparams)

   Bases: :class:`model.unet.Model`

   .. autoapi-inheritance-diagram:: unet_downsampled.Model
      :parts: 1

   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