Plonk / configs /model /von_fisher.yaml
nicolas-dufour's picture
squash: merge all unpushed commits
c4c7cee
raw
history blame contribute delete
442 Bytes
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