| _target_: pytorch_lightning.Trainer | |
| accelerator: 'gpu' | |
| devices: 1 | |
| num_nodes: 1 | |
| precision: 32 | |
| max_epochs: ${data_module.max_epochs} | |
| accumulate_grad_batches: 1 | |
| gradient_clip_val: 0.5 | |
| gradient_clip_algorithm: value | |
| check_val_every_n_epoch: 5 | |
| strategy: | |
| _target_: pytorch_lightning.strategies.ddp.DDPStrategy | |
| find_unused_parameters: true | |
| logger: | |
| _target_: pytorch_lightning.loggers.WandbLogger | |
| project: crystal-generation | |
| job_type: train | |
| settings: | |
| _target_: wandb.Settings | |
| start_method: fork | |
| _save_requirements: False | |
| callbacks: | |
| - _target_: pytorch_lightning.callbacks.LearningRateMonitor | |
| logging_interval: step | |
| log_momentum: False | |
| - _target_: pytorch_lightning.callbacks.ModelCheckpoint | |
| monitor: loss_val | |
| mode: min | |
| save_top_k: 1 | |
| save_last: True | |
| verbose: false | |
| every_n_epochs: 1 | |
| filename: "{epoch}-{loss_val:.2f}" | |
| - _target_: pytorch_lightning.callbacks.TQDMProgressBar | |
| refresh_rate: 50 | |
| - _target_: onescience.datapipes.materials.mattergen.callback.SetPropertyScalers | |