File tree Expand file tree Collapse file tree 8 files changed +684
-10
lines changed Expand file tree Collapse file tree 8 files changed +684
-10
lines changed Original file line number Diff line number Diff line change 1010=== "模型训练命令"
1111
1212 ``` sh
13- # ICAR-ENSO 数据预训练模型
13+ # ICAR-ENSO 数据模型训练
1414 python examples/earthformer/earthformer_enso_train.py
15- # SEVIR 数据集预训练模型
15+ # SEVIR 数据模型训练
1616 python examples/earthformer/earthformer_sevir_train.py
1717
1818 ```
@@ -74,7 +74,7 @@ EarthFormer 原代码中训练了 ICAR-ENSO 数据集中海面温度 (sst) 和 S
7474
7575### 2.1 ICAR-ENSO 和 SEVIR 模型的训练、推理过程
7676
77- 模型预训练阶段是基于随机初始化的网络权重对模型进行训练,如下图所示,其中 $[ x_ {i}] _ {i=1}^{T}$ 表示长度为 $T$ 时空序列的输入气象数据,$[ y_ {i}] _ {i=1}^{K}$ 表示预测未来 $K$ 步的气象数据,$[ y_ {i_True }] _ {i=1}^{K}$ 表示未来 $K$ 步的真实数据,如海面温度数据和云总降水量数据。最后网络模型预测的输出和真值计算 mse 损失函数。
77+ 模型预训练阶段是基于随机初始化的网络权重对模型进行训练,如下图所示,其中 $[ x_ {i}] _ {i=1}^{T}$ 表示长度为 $T$ 时空序列的输入气象数据,$[ y_ {T+ i}] _ {i=1}^{K}$ 表示预测未来 $K$ 步的气象数据,$[ y_ {T+i_true }] _ {i=1}^{K}$ 表示未来 $K$ 步的真实数据,如海面温度数据和云总降水量数据。最后网络模型预测的输出和真值计算 mse 损失函数。
7878
7979<figure markdown >
8080 ![ earthformer-pretraining] ( https://paddle-org.bj.bcebos.com/paddlescience/docs/earthformer/earthformer-pretrain.png ) { loading=lazy style="margin:0 auto;height:70%;width:70%"}
Load Diff Large diffs are not rendered by default.
Original file line number Diff line number Diff line change 4949 - Volterra_IDE : zh/examples/volterra_ide.md
5050 - NLSMB : zh/examples/nlsmb.md
5151 - XPINN : zh/examples/xpinns.md
52+ - NeuralOperator : zh/examples/neuraloperator.md
5253 - 技术科学(AI for Technology) :
5354 - 流体 :
5455 - AMGNet : zh/examples/amgnet.md
File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change 3535import paddle
3636from paddle import nn
3737
38- from ppsci .paddle_harmonics .sht import InverseRealSHT
38+ from ppsci .arch . paddle_harmonics .sht import InverseRealSHT
3939
4040
4141class GaussianRandomFieldS2 (nn .Layer ):
Original file line number Diff line number Diff line change 3939import paddle .fft
4040import paddle .nn as nn
4141
42- from ppsci .paddle_harmonics .legendre import _precompute_dlegpoly
43- from ppsci .paddle_harmonics .legendre import _precompute_legpoly
44- from ppsci .paddle_harmonics .quadrature import clenshaw_curtiss_weights
45- from ppsci .paddle_harmonics .quadrature import legendre_gauss_weights
46- from ppsci .paddle_harmonics .quadrature import lobatto_weights
42+ from ppsci .arch . paddle_harmonics .legendre import _precompute_dlegpoly
43+ from ppsci .arch . paddle_harmonics .legendre import _precompute_legpoly
44+ from ppsci .arch . paddle_harmonics .quadrature import clenshaw_curtiss_weights
45+ from ppsci .arch . paddle_harmonics .quadrature import legendre_gauss_weights
46+ from ppsci .arch . paddle_harmonics .quadrature import lobatto_weights
4747
4848
4949class RealSHT (nn .Layer ):
Original file line number Diff line number Diff line change 1010
1111from ppsci .arch import base
1212from ppsci .arch import fno_block
13- from ppsci .paddle_harmonics import sht as paddle_sht
13+ from ppsci .arch . paddle_harmonics import sht as paddle_sht
1414from ppsci .utils import initializer
1515
1616einsum_symbols = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
You can’t perform that action at this time.
0 commit comments