TR2-D2 / tr2d2-dna /configs_gosai /callbacks /checkpoint_every_n_steps.yaml
zyc4975matholic
Include DNA training code
303c2e0
raw
history blame contribute delete
381 Bytes
checkpoint_every_n_steps:
_target_: lightning.pytorch.callbacks.ModelCheckpoint
save_top_k: -1 # Do not save any "best" models; this callback is being used to save every n train steps
save_last: True # save model as ${save_dir}/checkpoints/last.ckpt
dirpath: ${checkpointing.save_dir}/checkpoints
verbose: True
auto_insert_metric_name: False
every_n_train_steps: 500