提交 8c244219 编写于 作者: D d00455729

net_parameter_init

上级 3e691e54
...@@ -424,6 +424,7 @@ def export(net, *inputs, file_name, file_format='GEIR'): ...@@ -424,6 +424,7 @@ def export(net, *inputs, file_name, file_format='GEIR'):
if is_training: if is_training:
net.set_train(mode=False) net.set_train(mode=False)
# export model # export model
net.init_parameters_data()
if file_format == 'GEIR': if file_format == 'GEIR':
_executor.compile(net, *inputs, phase='export') _executor.compile(net, *inputs, phase='export')
_executor.export(net, file_name, file_format) _executor.export(net, file_name, file_format)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册