fix remove meta in pynative mode error

4 years ago · 450f149ff4
parent 21d6722386
commit 450f149ff4
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):
        # 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):
+            if os.path.isfile(graph_file_name) and context.get_context("mode") == context.GRAPH_MODE:
                os.remove(graph_file_name)
            _save_graph(cb_params.train_network, graph_file_name)
            self._graph_saved = True