34
34
35
35
| 预训练模型 | 指标 |
36
36
| :--| :--|
37
- | [ fundiff_turbulence_mass_transfer_dit_pretrained.pdparams] ( https://paddle-org.bj.bcebos.com/paddlescience/models/fundiff/fundiff_turbulence_mass_transfer_dit_pretrained.pdparams ) | Mean relative p error: 0.0755 <br >Max relative p error: 0.1976 <br >Min relative p error: 0.0225 <br >Std relative p error: 0.0389 <br >Mean relative sdf error: 0.0841 <br >Max relative sdf error: 0.2972 <br >Min relative sdf error: 0.0205 <br >Std relative sdf error: 0.0609 |
37
+ | [ fundiff_turbulence_mass_transfer_dit_pretrained.pdparams] ( https://paddle-org.bj.bcebos.com/paddlescience/models/fundiff/fundiff_turbulence_mass_transfer_dit_pretrained.pdparams ) | Mean relative p error: 0.0651 <br >Max relative p error: 0.1329 <br >Min relative p error: 0.0345 <br >Std relative p error: 0.0300 <br >Mean relative sdf error: 0.0684 <br >Max relative sdf error: 0.1443 <br >Min relative sdf error: 0.0335 <br >Std relative sdf error: 0.0377 |
38
38
39
39
## 1. 背景简介
40
40
55
55
56
56
在 FuncDiff 模型中,FAE 模块采用了 Perceiver 的架构,其输入为物理场 $x$ 和查询坐标 $coords$,输出是某个物理场在查询坐标上的值 $u$,因此模型构建代码如下
57
57
58
- ``` yaml linenums="36"
58
+ ``` yaml linenums="36" title="fae.yaml"
59
59
--8<--
60
60
examples/fundiff/conf/fae.yaml:36:63
61
61
--8<--
62
62
```
63
63
64
- ``` py linenums="87"
64
+ ``` py linenums="87" title="main.py"
65
65
-- 8 < --
66
66
examples/ fundiff/ main.py:87 :95
67
67
-- 8 < --
@@ -71,7 +71,7 @@ examples/fundiff/main.py:87:95
71
71
72
72
FAE 使用 auto encoder decoder 的训练范式,因此标签即是输入 $u$
73
73
74
- ``` py linenums="97"
74
+ ``` py linenums="97" title="main.py"
75
75
-- 8 < --
76
76
examples/ fundiff/ main.py:97 :148
77
77
-- 8 < --
@@ -83,13 +83,13 @@ examples/fundiff/main.py:97:148
83
83
84
84
DiT 的模型构建如下
85
85
86
- ``` yaml linenums="66"
86
+ ``` yaml linenums="66" title="diffusion.yaml"
87
87
--8<--
88
88
examples/fundiff/conf/diffusion.yaml:66:76
89
89
--8<--
90
90
```
91
91
92
- ``` py linenums="186"
92
+ ``` py linenums="186" title="main.py"
93
93
-- 8 < --
94
94
examples/ fundiff/ main.py:186 :208
95
95
-- 8 < --
105
105
106
106
其对应的前向计算实现代码如下
107
107
108
- ``` py linenums="44"
108
+ ``` py linenums="44" title="main.py"
109
109
-- 8 < --
110
110
examples/ fundiff/ main.py:44 :85
111
111
-- 8 < --
112
112
```
113
113
114
114
整体约束构建如下
115
115
116
- ``` py linenums="210"
116
+ ``` py linenums="210" title="main.py"
117
117
-- 8 < --
118
118
examples/ fundiff/ main.py:210 :262
119
119
-- 8 < --
@@ -147,7 +147,7 @@ examples/fundiff/conf/fae.yaml:65:80
147
147
--8<--
148
148
```
149
149
150
- ``` py linenums="159"
150
+ ``` py linenums="159" title="main.py"
151
151
-- 8 < --
152
152
examples/ fundiff/ main.py:159 :172
153
153
-- 8 < --
0 commit comments