|
|
|
@@ -481,8 +481,8 @@ def _save_graph(network, file_name): |
|
|
|
graph_pb = network.get_func_graph_proto() |
|
|
|
if graph_pb: |
|
|
|
with open(file_name, "wb") as f: |
|
|
|
os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR) |
|
|
|
f.write(graph_pb) |
|
|
|
os.chmod(file_name, stat.S_IRUSR) |
|
|
|
|
|
|
|
|
|
|
|
def _get_merged_param_data(net, param_name, param_data, integrated_save): |
|
|
|
@@ -637,7 +637,7 @@ def _export(net, file_name, file_format, *inputs): |
|
|
|
if not file_name.endswith('.onnx'): |
|
|
|
file_name += ".onnx" |
|
|
|
with open(file_name, 'wb') as f: |
|
|
|
os.chmod(file_name, stat.S_IWUSR | stat.S_IRUSR) |
|
|
|
os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR) |
|
|
|
f.write(onnx_stream) |
|
|
|
elif file_format == 'MINDIR': |
|
|
|
_save_mindir(net, file_name, *inputs) |
|
|
|
@@ -687,7 +687,7 @@ def _save_mindir(net, file_name, *inputs): |
|
|
|
dirname = os.path.dirname(current_path) |
|
|
|
os.makedirs(dirname, exist_ok=True) |
|
|
|
with open(file_name, 'wb') as f: |
|
|
|
os.chmod(file_name, stat.S_IWUSR | stat.S_IRUSR) |
|
|
|
os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR) |
|
|
|
f.write(model.SerializeToString()) |
|
|
|
else: |
|
|
|
logger.warning("Parameters in the net capacity exceeds 1G, save MindIR model and parameters separately.") |
|
|
|
@@ -701,6 +701,7 @@ def _save_mindir(net, file_name, *inputs): |
|
|
|
if os.path.exists(data_path): |
|
|
|
shutil.rmtree(data_path) |
|
|
|
os.makedirs(data_path, exist_ok=True) |
|
|
|
os.chmod(data_path, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR) |
|
|
|
index = 0 |
|
|
|
graphproto = graph_proto() |
|
|
|
data_size = 0 |
|
|
|
@@ -720,6 +721,7 @@ def _save_mindir(net, file_name, *inputs): |
|
|
|
if data_size > TOTAL_SAVE: |
|
|
|
data_file_name = data_path + "/" + "data_" + str(index) |
|
|
|
with open(data_file_name, "ab") as f: |
|
|
|
os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR) |
|
|
|
f.write(graphproto.SerializeToString()) |
|
|
|
index += 1 |
|
|
|
data_size = 0 |
|
|
|
@@ -728,13 +730,14 @@ def _save_mindir(net, file_name, *inputs): |
|
|
|
if graphproto.parameter: |
|
|
|
data_file_name = data_path + "/" + "data_" + str(index) |
|
|
|
with open(data_file_name, "ab") as f: |
|
|
|
os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR) |
|
|
|
f.write(graphproto.SerializeToString()) |
|
|
|
|
|
|
|
# save graph |
|
|
|
del model.graph.parameter[:] |
|
|
|
graph_file_name = dirname + "/" + file_prefix + "_graph.mindir" |
|
|
|
with open(graph_file_name, 'wb') as f: |
|
|
|
os.chmod(graph_file_name, stat.S_IWUSR | stat.S_IRUSR) |
|
|
|
os.chmod(graph_file_name, stat.S_IRUSR | stat.S_IWUSR) |
|
|
|
f.write(model.SerializeToString()) |
|
|
|
|
|
|
|
|
|
|
|
|