configs/model/von_fisher.yaml

defaults:
  - optimizer: lamb
  - lr_scheduler: warmup_cosine_decay
  - network: geo_adaln_mlp_von_fisher
  - preconditioning: ddpm
  - data_preprocessing: gps_to_cartesian
  - cond_preprocessing: embedding
  - postprocessing: cartesian_to_gps
  - loss: von_fisher
  - val_sampler: von_fisher
  - test_sampler: von_fisher
  - _self_

network:
  input_dim: 3
name: GeoMLP_R3_VonFisher
ema_decay: 0.999
start_ema_step: 0
interpolant: von_fisher