Browse Source

fix remove meta in pynative mode error

tags/v1.2.0-rc1
caozhou 5 years ago
parent
commit
450f149ff4
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):
# 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


Loading…
Cancel
Save