!13806 add air restriction note

From: @caozhou_huawei
Reviewed-by: @kingxian,@zh_qh
Signed-off-by: @kingxian
This commit is contained in:
mindspore-ci-bot 2021-03-23 16:59:19 +08:00 committed by Gitee
commit c4e339674b
1 changed files with 4 additions and 0 deletions

View File

@ -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`.