File size: 1,037 Bytes
f15d29e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | _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
|