Skip to content

Commit 53c4ede

Browse files
[Hackathon 6th Code Camp No.15] support neuraloperator docs (#917)
* add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator * add-neuraloperator-doc * add-neuraloperator-doc * move-paddle-harmonics * add-neuraloperator-doc * add-neuraloperator-doc * add-neuraloperator-doc * add-neuraloperator-doc * add-neuraloperator-doc * add-neuraloperator-doc * add-neuraloperator-docs * add-neuraloperator-docs * add-neuraloperator-docs * add-neuraloperator-doc * add-neuraloperator-doc
1 parent cc43efa commit 53c4ede

File tree

8 files changed

+684
-10
lines changed

8 files changed

+684
-10
lines changed

docs/zh/examples/earthformer.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,9 @@
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%"}

docs/zh/examples/neuraloperator.md

Lines changed: 673 additions & 0 deletions
Large diffs are not rendered by default.

mkdocs.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ nav:
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

ppsci/paddle_harmonics/random_fields.py renamed to ppsci/arch/paddle_harmonics/random_fields.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
import paddle
3636
from paddle import nn
3737

38-
from ppsci.paddle_harmonics.sht import InverseRealSHT
38+
from ppsci.arch.paddle_harmonics.sht import InverseRealSHT
3939

4040

4141
class GaussianRandomFieldS2(nn.Layer):

ppsci/paddle_harmonics/sht.py renamed to ppsci/arch/paddle_harmonics/sht.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,11 @@
3939
import paddle.fft
4040
import 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

4949
class RealSHT(nn.Layer):

ppsci/arch/sfnonet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
from ppsci.arch import base
1212
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
1414
from ppsci.utils import initializer
1515

1616
einsum_symbols = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"

0 commit comments

Comments
 (0)