From cab7b623f565d7c672da448733096b0e7fe60100 Mon Sep 17 00:00:00 2001 From: caozhou Date: Tue, 9 Mar 2021 16:30:25 +0800 Subject: [PATCH] add ckpt directory note --- mindspore/train/callback/_checkpoint.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mindspore/train/callback/_checkpoint.py b/mindspore/train/callback/_checkpoint.py index 272dbe43e9..a8c36c5b6a 100644 --- a/mindspore/train/callback/_checkpoint.py +++ b/mindspore/train/callback/_checkpoint.py @@ -221,7 +221,11 @@ class ModelCheckpoint(Callback): """ The checkpoint callback class. - It is called to combine with train process and save the model and network parameters after traning. + It is called to combine with train process and save the model and network parameters after training. + + Note: + In the distributed training scenario, please specify different directories for each training process + to save the checkpoint file. Otherwise, the training may fail. Args: prefix (str): The prefix name of checkpoint files. Default: "CKP".