From c27ecbcdae3a66cba033d6af4632cae0d677a90b Mon Sep 17 00:00:00 2001 From: caozhou Date: Tue, 23 Mar 2021 10:35:14 +0800 Subject: [PATCH] add air restriction note --- mindspore/train/serialization.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index 4615034015..dd1076a094 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -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`.