Browse Source

add ckpt directory note

tags/v1.2.0-rc1
caozhou 5 years ago
parent
commit
cab7b623f5
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      mindspore/train/callback/_checkpoint.py

+ 5
- 1
mindspore/train/callback/_checkpoint.py View File

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


Loading…
Cancel
Save