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".