|
|
@ -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)
|
|
|
|