Browse Source

!6789 fix int in ckpt name changed to float bug

Merge pull request !6789 from caozhou/fix_ckpt_name_float
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
04e2ab1b0b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/callback/_checkpoint.py

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

@@ -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) + "_" \


Loading…
Cancel
Save