From be55197617110079aa280a839b4bc81280780589 Mon Sep 17 00:00:00 2001 From: xingzhongfan Date: Tue, 1 Apr 2025 17:23:53 +0800 Subject: [PATCH] fix reweitten netword dict --- MindFlow/applications/data_driven/burgers/fno1d/FNO1D.ipynb | 2 +- MindFlow/applications/data_driven/burgers/fno1d/FNO1D_CN.ipynb | 2 +- MindFlow/applications/data_driven/burgers/fno1d/train.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/MindFlow/applications/data_driven/burgers/fno1d/FNO1D.ipynb b/MindFlow/applications/data_driven/burgers/fno1d/FNO1D.ipynb index a42b0a942..ca19d6027 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 0c82707cf..539697fc9 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 9fc2cb039..4acc92fb8 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 -- Gitee