net_parameter_init

5 years ago · 8c2442192a
parent 3e691e54f5
commit 8c2442192a
1 changed files with 1 additions and 0 deletions
--- a/mindspore/train/serialization.py
+++ b/mindspore/train/serialization.py
@ -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)