fix int in ckpt name changed to float bug

4 years ago · 5221041490
parent 8d20d1b8e6
commit 5221041490
1 changed files with 1 additions and 1 deletions
--- a/mindspore/train/callback/_checkpoint.py
+++ b/mindspore/train/callback/_checkpoint.py
@ -276,7 +276,7 @@ class ModelCheckpoint(Callback):
            return

        save_ckpt = self._check_save_ckpt(cb_params, force_to_save)
-        step_num_in_epoch = (cb_params.cur_step_num - 1) % cb_params.batch_num + 1
+        step_num_in_epoch = int((cb_params.cur_step_num - 1) % cb_params.batch_num + 1)

        if save_ckpt:
            cur_ckpoint_file = self._prefix + "-" + str(cb_params.cur_epoch_num) + "_" \