From 522104149076432097f8d6f012cbf7fe1c7779af Mon Sep 17 00:00:00 2001 From: caozhou Date: Wed, 23 Sep 2020 17:09:54 +0800 Subject: [PATCH] fix int in ckpt name changed to float bug --- mindspore/train/callback/_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/train/callback/_checkpoint.py b/mindspore/train/callback/_checkpoint.py index ce3892a368..030fafbea8 100644 --- 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) + "_" \