Olamide Isreal
a3ffec6a07
Some checks failed
Code Quality Main / code-quality (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.10) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.8) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.9) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.10) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.8) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.9) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.10) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.8) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.9) (push) Has been cancelled
Tests / code-coverage (push) Has been cancelled
19 lines
1.3 KiB
YAML
19 lines
1.3 KiB
YAML
# https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.callbacks.ModelCheckpoint.html
|
||
|
||
model_checkpoint:
|
||
_target_: flowdock.models.components.callbacks.ema.EMAModelCheckpoint
|
||
dirpath: null # directory to save the model file
|
||
filename: "best" # checkpoint filename
|
||
monitor: val_sampling/ligand_hit_score_2A_epoch # name of the logged metric which determines when model is improving
|
||
verbose: True # verbosity mode
|
||
save_last: False # additionally always save an exact copy of the last checkpoint to a file last.ckpt
|
||
save_top_k: 1 # save k best models (determined by above metric)
|
||
mode: "max" # "max" means higher metric value is better, can be also "min"
|
||
auto_insert_metric_name: True # when True, the checkpoints filenames will contain the metric name
|
||
save_weights_only: False # if True, then only the model’s weights will be saved
|
||
every_n_train_steps: null # number of training steps between checkpoints
|
||
train_time_interval: null # checkpoints are monitored at the specified time interval
|
||
every_n_epochs: null # number of epochs between checkpoints
|
||
save_on_train_epoch_end: null # whether to run checkpointing at the end of the training epoch or the end of validation
|
||
enable_version_counter: False # enables versioning for checkpoint names
|