add air restriction note

pull/13806/head
caozhou 4 years ago
parent 1ad9f9de21
commit c27ecbcdae

@ -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. 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: Args:
net (Cell): MindSpore network. net (Cell): MindSpore network.
inputs (Tensor): Inputs of the `net`. inputs (Tensor): Inputs of the `net`.

Loading…
Cancel
Save