delete meta

2020-12-26 17:03:08 +08:00 · 2020-12-26 17:03:08 +08:00 · 08a13ab71e
parent 8927f2d644
commit 08a13ab71e
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)