model:
  name: SNO2D
  in_channels: 3
  out_channels: 1
  hidden_channels: 32
  num_sno_layers: 2
  kernel_size: 5
  num_usno_layers: 1
  num_unet_strides: 1
  resolution: 64
  modes: 12
  activation: gelu