!10612 delete meta

From: @caozhou_huawei Reviewed-by: @changzherui,@zh_qh,@zhunaipan Signed-off-by: @zhunaipan
2020-12-27 09:42:00 +08:00 · 2020-12-27 09:42:00 +08:00 · dd3fd238e0
parent 5653a0a970 08a13ab71e
commit dd3fd238e0
1 changed files with 2 additions and 0 deletions
--- a/mindspore/train/callback/_checkpoint.py
+++ b/mindspore/train/callback/_checkpoint.py
@ -275,6 +275,8 @@ class ModelCheckpoint(Callback):
        # save graph (only once)
        if not self._graph_saved:
            graph_file_name = os.path.join(self._directory, self._prefix + '-graph.meta')
+            if os.path.isfile(graph_file_name):
+                os.remove(graph_file_name)
            _save_graph(cb_params.train_network, graph_file_name)
            self._graph_saved = True
        self._save_ckpt(cb_params)