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 10
10
=== "模型训练命令"
11
11
12
12
``` sh
13
- # ICAR-ENSO 数据预训练模型
13
+ # ICAR-ENSO 数据模型训练
14
14
python examples/earthformer/earthformer_enso_train.py
15
- # SEVIR 数据集预训练模型
15
+ # SEVIR 数据模型训练
16
16
python examples/earthformer/earthformer_sevir_train.py
17
17
18
18
```
@@ -74,7 +74,7 @@ EarthFormer 原代码中训练了 ICAR-ENSO 数据集中海面温度 (sst) 和 S
74
74
75
75
### 2.1 ICAR-ENSO 和 SEVIR 模型的训练、推理过程
76
76
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 损失函数。
78
78
79
79
<figure markdown >
80
80
![ 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 49
49
- Volterra_IDE : zh/examples/volterra_ide.md
50
50
- NLSMB : zh/examples/nlsmb.md
51
51
- XPINN : zh/examples/xpinns.md
52
+ - NeuralOperator : zh/examples/neuraloperator.md
52
53
- 技术科学(AI for Technology) :
53
54
- 流体 :
54
55
- AMGNet : zh/examples/amgnet.md
File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change 35
35
import paddle
36
36
from paddle import nn
37
37
38
- from ppsci .paddle_harmonics .sht import InverseRealSHT
38
+ from ppsci .arch . paddle_harmonics .sht import InverseRealSHT
39
39
40
40
41
41
class GaussianRandomFieldS2 (nn .Layer ):
Original file line number Diff line number Diff line change 39
39
import paddle .fft
40
40
import paddle .nn as nn
41
41
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
47
47
48
48
49
49
class RealSHT (nn .Layer ):
Original file line number Diff line number Diff line change 10
10
11
11
from ppsci .arch import base
12
12
from 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
14
14
from ppsci .utils import initializer
15
15
16
16
einsum_symbols = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
You can’t perform that action at this time.
0 commit comments