Hugging Face's logo

OneScience-Group
/
MP_PDE

English
OneScience
fluid-mechanics
flow-field-prediction
graph-neural-network
partial-differential-equations
Model card Files
xet
 Community
yushuang88 commited on
Commit
d15f55c
·
verified ·
1 Parent(s): 90e1458

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +25 -179
config.json CHANGED
@@ -1,185 +1,31 @@
1
  {
2
- "architectures": [
3
- "MPPDESolver"
 
 
 
 
 
 
 
 
4
  ],
5
- "framework": "pytorch",
6
- "model_name": "MP_PDE",
7
- "model_type": "mp_pde",
8
- "source_config": "config/config.yaml",
9
- "experiment": {
10
- "name": "E3",
11
- "model": "MP-PDE",
12
- "paper": "https://arxiv.org/abs/2202.03376",
13
- "evidence_priority": "paper_then_official_repo_for_missing_details",
14
- "ambiguity_policy": "paper_strict"
15
- },
16
- "paths": {
17
- "data": "data/e3.h5",
18
- "checkpoint": "weight/best_model.pth",
19
- "results": "results",
20
- "train_history": "results/train_history.json",
21
- "predictions": "results/predictions.npz",
22
- "metrics": "results/metrics.json"
23
- },
24
- "data": {
25
- "domain_length": 16.0,
26
- "final_time": 4.0,
27
- "num_time_points": 250,
28
- "high_resolution_nx": 200,
29
- "resolution": 100,
30
- "supported_resolutions": [
31
- 100,
32
- 50,
33
- 40
34
- ],
35
- "sample_fraction": 0.3333333333333333,
36
- "full_experiment_samples": {
37
- "train": 2096,
38
- "valid": 128,
39
- "test": 128
40
- },
41
- "train_samples": 698,
42
- "valid_samples": 43,
43
- "test_samples": 43,
44
- "seed": 0,
45
- "equation": {
46
- "alpha_range": [
47
- 0.0,
48
- 3.0
49
- ],
50
- "beta_range": [
51
- 0.0,
52
- 0.4
53
- ],
54
- "gamma_range": [
55
- 0.0,
56
- 1.0
57
- ]
58
- },
59
- "forcing": {
60
- "terms": 5,
61
- "amplitude_range": [
62
- -0.5,
63
- 0.5
64
- ],
65
- "paper_omega_range": [
66
- -0.4,
67
- -0.4
68
- ],
69
- "official_consistency_omega_range": [
70
- -0.4,
71
- 0.4
72
- ],
73
- "modes": [
74
- 1,
75
- 2,
76
- 3
77
- ],
78
- "phase_range": [
79
- 0.0,
80
- 6.283185307179586
81
- ]
82
- },
83
- "generation": {
84
- "cfl": 0.2,
85
- "min_dt": 1e-08,
86
- "max_substeps": 10000,
87
- "weno_epsilon": 1e-06,
88
- "compression": "gzip",
89
- "compression_level": 4
90
- },
91
- "parallel_generation": {
92
- "workers": 8,
93
- "max_in_flight": 16,
94
- "flush_every": 8,
95
- "resume_partial": true
96
- }
97
- },
98
- "model": {
99
- "time_window": 25,
100
- "hidden_dim": 164,
101
- "message_passing_layers": 6,
102
- "neighbor_offsets": [
103
- -3,
104
- -2,
105
- -1,
106
- 1,
107
- 2,
108
- 3
109
- ],
110
- "aggregation": "sum",
111
- "instance_norm_affine": false,
112
- "decoder": {
113
- "middle_channels": 8,
114
- "kernel_sizes": [
115
- 16,
116
- 26
117
- ],
118
- "strides": [
119
- 3,
120
- 1
121
- ]
122
- },
123
- "scaling": {
124
- "coordinates": true,
125
- "parameters": true,
126
- "solution": "none"
127
- }
128
- },
129
- "training": {
130
- "epochs": 20,
131
- "batch_size": 16,
132
- "num_workers": 0,
133
- "optimizer": "AdamW",
134
- "learning_rate": 0.0001,
135
- "weight_decay": 1e-08,
136
- "loss": "rmse",
137
- "loss_epsilon": 1e-12,
138
- "max_no_grad_pushes": 1,
139
- "scheduler": {
140
- "enabled": true,
141
- "milestones": [
142
- 1,
143
- 5,
144
- 10,
145
- 15
146
- ],
147
- "gamma": 0.4
148
- },
149
- "precision": "float32",
150
- "amp": false,
151
- "gradient_clip": null,
152
- "seed": 0,
153
- "deterministic": true,
154
- "device": "auto",
155
- "print_interval": 10,
156
- "validation_rollout_samples": null,
157
- "test_rollout_samples": null
158
- },
159
- "inference": {
160
- "batch_size": 16,
161
- "split": "test",
162
- "max_samples": null,
163
- "device": "auto",
164
- "forecast_start_index": 25
165
- },
166
- "visualization": {
167
- "sample_index": 0,
168
- "time_indices": [
169
- 25,
170
- 75,
171
- 150,
172
- 249
173
  ],
174
- "dpi": 160
 
 
 
175
  },
176
- "paper_reference": {
177
- "metric": "forecast_accumulated_mse",
178
- "values": {
179
- "100": 4.26,
180
- "50": 3.74,
181
- "40": 3.7
182
- },
183
- "note": "provenance_only_not_an_assertion"
184
  }
185
  }
 
1
  {
2
+ "time_window": 25,
3
+ "hidden_dim": 164,
4
+ "message_passing_layers": 6,
5
+ "neighbor_offsets": [
6
+ -3,
7
+ -2,
8
+ -1,
9
+ 1,
10
+ 2,
11
+ 3
12
  ],
13
+ "aggregation": "sum",
14
+ "instance_norm_affine": false,
15
+ "decoder": {
16
+ "middle_channels": 8,
17
+ "kernel_sizes": [
18
+ 16,
19
+ 26
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  ],
21
+ "strides": [
22
+ 3,
23
+ 1
24
+ ]
25
  },
26
+ "scaling": {
27
+ "coordinates": true,
28
+ "parameters": true,
29
+ "solution": "none"
 
 
 
 
30
  }
31
  }