add air restriction note

2021-03-23 10:35:14 +08:00 · 2021-03-23 10:35:14 +08:00 · c27ecbcdae
parent 1ad9f9de21
commit c27ecbcdae
1 changed files with 4 additions and 0 deletions
--- 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`.