From 29857484439553bd7165ac10c1c644fdf0b8a81c Mon Sep 17 00:00:00 2001 From: talregev Date: Wed, 6 Apr 2022 21:46:53 +0300 Subject: [PATCH 1/2] [Docs] Example for auto_insert_metric_name Example for auto_insert_metric_name: - Add an example in case you have slash in metric name --- pytorch_lightning/callbacks/model_checkpoint.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py index 223e2639d9c68..ebde306ad23f2 100644 --- a/pytorch_lightning/callbacks/model_checkpoint.py +++ b/pytorch_lightning/callbacks/model_checkpoint.py @@ -103,6 +103,7 @@ class ModelCheckpoint(Callback): For example, ``filename='checkpoint_{epoch:02d}-{acc:02.0f}`` with epoch ``1`` and acc ``1.12`` will resolve to ``checkpoint_epoch=01-acc=01.ckpt``. Is useful to set it to ``False`` when metric names contain ``/`` as this will result in extra folders. + For example, `filename='epoch={epoch}-step={step}-val_acc={val/acc:.2f}', auto_insert_metric_name=False` save_weights_only: if ``True``, then only the model's weights will be saved. Otherwise, the optimizer states, lr-scheduler states, etc are added in the checkpoint too. every_n_train_steps: Number of training steps between checkpoints. From ef55c425ffa19d47b8dca6b3f50c62e3576f1555 Mon Sep 17 00:00:00 2001 From: talregev Date: Mon, 18 Apr 2022 21:15:14 +0300 Subject: [PATCH 2/2] Update pytorch_lightning/callbacks/model_checkpoint.py Co-authored-by: Jirka Borovec --- pytorch_lightning/callbacks/model_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py index ebde306ad23f2..9ef5b856dacaa 100644 --- a/pytorch_lightning/callbacks/model_checkpoint.py +++ b/pytorch_lightning/callbacks/model_checkpoint.py @@ -103,7 +103,7 @@ class ModelCheckpoint(Callback): For example, ``filename='checkpoint_{epoch:02d}-{acc:02.0f}`` with epoch ``1`` and acc ``1.12`` will resolve to ``checkpoint_epoch=01-acc=01.ckpt``. Is useful to set it to ``False`` when metric names contain ``/`` as this will result in extra folders. - For example, `filename='epoch={epoch}-step={step}-val_acc={val/acc:.2f}', auto_insert_metric_name=False` + For example, ``filename='epoch={epoch}-step={step}-val_acc={val/acc:.2f}', auto_insert_metric_name=False`` save_weights_only: if ``True``, then only the model's weights will be saved. Otherwise, the optimizer states, lr-scheduler states, etc are added in the checkpoint too. every_n_train_steps: Number of training steps between checkpoints.