yushuang88 commited on
Commit
ccf5efb
·
verified ·
1 Parent(s): e1b9c0f

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +67 -0
config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "GFNO"
4
+ ],
5
+ "framework": "pytorch",
6
+ "model_name": "GFNO",
7
+ "model_type": "gfno",
8
+ "source_config": "conf/config.yaml",
9
+ "root": {
10
+ "common": {
11
+ "device": "auto",
12
+ "seed": 0
13
+ },
14
+ "datapipe": {
15
+ "source": {
16
+ "data_dir": "/public/share/sugonhpcapp01/onestore/onedatasets/CFD_Benchmark/ns",
17
+ "file_name": "NavierStokes_V1e-5_N1200_T20.mat"
18
+ },
19
+ "data": {
20
+ "ntrain": 1000,
21
+ "ntest": 200,
22
+ "t_in": 10,
23
+ "t_out": 10,
24
+ "out_dim": 1,
25
+ "downsamplex": 1,
26
+ "downsampley": 1,
27
+ "normalize": false
28
+ },
29
+ "dataloader": {
30
+ "batch_size": 8,
31
+ "num_workers": 4,
32
+ "pin_memory": true
33
+ }
34
+ },
35
+ "model": {
36
+ "in_dim": 10,
37
+ "out_dim": 1,
38
+ "hidden_dim": 64,
39
+ "modes": 12,
40
+ "num_layers": 4,
41
+ "space_dim": 2,
42
+ "include_pos": true,
43
+ "activation": "gelu",
44
+ "reflection": false,
45
+ "pad_to_multiple": 16
46
+ },
47
+ "training": {
48
+ "epochs": 500,
49
+ "lr": 0.001,
50
+ "weight_decay": 1e-05,
51
+ "step_size": 100,
52
+ "gamma": 0.5,
53
+ "eval_interval": 1,
54
+ "patience": 100,
55
+ "teacher_forcing": false,
56
+ "max_grad_norm": null,
57
+ "max_train_batches": null,
58
+ "max_eval_batches": null,
59
+ "weight_dir": "weight",
60
+ "checkpoint_name": "gfno_navier_stokes.pt"
61
+ },
62
+ "inference": {
63
+ "result_dir": "result",
64
+ "num_samples": 4
65
+ }
66
+ }
67
+ }