model:
  name: SNO3D
  in_channels: 4
  out_channels: 1
  hidden_channels: 16
  num_sno_layers: 2
  kernel_size: 3
  resolution_xy: 32
  resolution_z: 24
  modes: 8
  activation: gelu