simergy-diffusion-tmo-alpha / training_overrides.yaml
HishaamA's picture
Upload Simergy diffusion TMO alpha checkpoint
8de8b02 verified
Raw
History Blame Contribute Delete
242 Bytes
- trainer=single_gpu
- data_module=tmo_ehull
- adapter.pretrained_name=chemical_system_energy_above_hull
- adapter.full_finetuning=true
- trainer.max_epochs=100
- trainer.accumulate_grad_batches=8
- lightning_module.optimizer_partial.lr=5e-6