dtc_super_resolution.rcm_emulator.ml_module.basic_unet.unet

Full assembly of the parts to form the complete network.

Classes

UNetBaseline([n_channels_x, n_channels_z, ...])

UNet model for super-resolution of climate data.

class UNetBaseline(n_channels_x: int = 8, n_channels_z: int = 18, size: int = 64, n_filter: int = 64, bilinear: bool = False)

Bases: Module

UNet model for super-resolution of climate data.

forward(x: Tensor, z: Tensor) → Tensor

Forward pass for the UNetBaseline.