diff --git a/MindFlow/applications/data_driven/burgers/fno1d/FNO1D.ipynb b/MindFlow/applications/data_driven/burgers/fno1d/FNO1D.ipynb
index a42b0a94246477da0d7bcf83fbbc5f2ed94201ee..ca19d6027acef1da65c3dc8d482727b852fc8e76 100644
--- a/MindFlow/applications/data_driven/burgers/fno1d/FNO1D.ipynb
+++ b/MindFlow/applications/data_driven/burgers/fno1d/FNO1D.ipynb
@@ -310,7 +310,7 @@
                 "if use_ascend:\n",
                 "    from mindspore.amp import DynamicLossScaler, auto_mixed_precision, all_finite\n",
                 "    loss_scaler = DynamicLossScaler(1024, 2, 100)\n",
-                "    auto_mixed_precision(model, 'O1')\n",
+                "    auto_mixed_precision(model, 'O3')\n",
                 "else:\n",
                 "    loss_scaler = None"
             ]
diff --git a/MindFlow/applications/data_driven/burgers/fno1d/FNO1D_CN.ipynb b/MindFlow/applications/data_driven/burgers/fno1d/FNO1D_CN.ipynb
index 0c82707cf951f8ca34bad4abbd12e6eefdcf698f..539697fc94b289646ac453bc06406f35f5bc9af9 100644
--- a/MindFlow/applications/data_driven/burgers/fno1d/FNO1D_CN.ipynb
+++ b/MindFlow/applications/data_driven/burgers/fno1d/FNO1D_CN.ipynb
@@ -316,7 +316,7 @@
                 "if use_ascend:\n",
                 "    from mindspore.amp import DynamicLossScaler, auto_mixed_precision, all_finite\n",
                 "    loss_scaler = DynamicLossScaler(1024, 2, 100)\n",
-                "    auto_mixed_precision(model, 'O1')\n",
+                "    auto_mixed_precision(model, 'O3')\n",
                 "else:\n",
                 "    loss_scaler = None"
             ]
diff --git a/MindFlow/applications/data_driven/burgers/fno1d/train.py b/MindFlow/applications/data_driven/burgers/fno1d/train.py
index 9fc2cb039fb9a9cfd237f5012872b11f8dce1b65..4acc92fb889a8119670719bbd6fb06b75cf43cc4 100644
--- a/MindFlow/applications/data_driven/burgers/fno1d/train.py
+++ b/MindFlow/applications/data_driven/burgers/fno1d/train.py
@@ -87,7 +87,7 @@ def train(input_args):
     if use_ascend:
         from mindspore.amp import DynamicLossScaler, auto_mixed_precision, all_finite
         loss_scaler = DynamicLossScaler(1024, 2, 100)
-        auto_mixed_precision(model, 'O1')
+        auto_mixed_precision(model, 'O3')
     else:
         loss_scaler = None