forked from mindspore-Ecosystem/mindspore
!13806 add air restriction note
From: @caozhou_huawei Reviewed-by: @kingxian,@zh_qh Signed-off-by: @kingxian
This commit is contained in:
commit
c4e339674b
|
@ -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`.
|
||||
|
|
Loading…
Reference in New Issue