diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py
index 46150340150..dd1076a0944 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`.