|
|
|
@ -575,6 +575,10 @@ def export(net, *inputs, file_name, file_format='AIR', **kwargs):
|
|
|
|
|
"""
|
|
|
|
|
Export the MindSpore prediction model to a file in the specified format.
|
|
|
|
|
|
|
|
|
|
Notes:
|
|
|
|
|
When exporting to AIR format, the size of a single tensor can not exceed 2GB.
|
|
|
|
|
Otherwise, a small wrong air file will be exported.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
net (Cell): MindSpore network.
|
|
|
|
|
inputs (Tensor): Inputs of the `net`.
|
|
|
|
|