Browse Source

!2482 fix a bug :serialization.export parameter "file_name" doesn't work

Merge pull request !2482 from lilei/sparse_proximal_ada_grad
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
002bddecfd
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/common/api.py

+ 1
- 1
mindspore/common/api.py View File

@@ -523,7 +523,7 @@ class _Executor:
file_format (str): MindSpore currently support 'GEIR' and 'ONNX' format for exported model
"""
from .._c_expression import export_graph
phase = 'export' + '.' + str(net.create_time)
phase = 'export' + '.' + self.phase_prefix + '.' + str(net.create_time)
export_graph(file_name, file_format, phase)

def fetch_info_for_quant_export(self, exec_id):


Loading…
Cancel
Save