Browse Source

net_parameter_init

tags/v0.6.0-beta
d00455729 5 years ago
parent
commit
8c2442192a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/train/serialization.py

+ 1
- 0
mindspore/train/serialization.py View File

@@ -424,6 +424,7 @@ def export(net, *inputs, file_name, file_format='GEIR'):
if is_training:
net.set_train(mode=False)
# export model
net.init_parameters_data()
if file_format == 'GEIR':
_executor.compile(net, *inputs, phase='export')
_executor.export(net, file_name, file_format)


Loading…
Cancel
Save