Browse Source

!13039 add ckpt direcoty note

From: @caozhou_huawei
Reviewed-by: @kingxian,@zh_qh
Signed-off-by: @kingxian
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
c0a4680ce0
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