Browse Source

!13806 add air restriction note

From: @caozhou_huawei
Reviewed-by: @kingxian,@zh_qh
Signed-off-by: @kingxian
pull/13806/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
c4e339674b
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/train/serialization.py

+ 4
- 0
mindspore/train/serialization.py 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`.


Loading…
Cancel
Save