yushuang88 commited on
Commit
a404e09
·
verified ·
1 Parent(s): 2ee8970

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +66 -0
config.json ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "KNO2DNavierStokes"
4
+ ],
5
+ "framework": "pytorch",
6
+ "model_name": "KNO",
7
+ "model_type": "kno",
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
+ "name": "KNO2DNavierStokes",
37
+ "input_channels": 10,
38
+ "output_channels": 1,
39
+ "op_size": 32,
40
+ "modes_x": 12,
41
+ "modes_y": 12,
42
+ "decompose": 6,
43
+ "linear_type": true,
44
+ "normalization": false
45
+ },
46
+ "training": {
47
+ "epochs": 500,
48
+ "lr": 0.001,
49
+ "weight_decay": 1e-05,
50
+ "step_size": 100,
51
+ "gamma": 0.5,
52
+ "eval_interval": 10,
53
+ "patience": 100,
54
+ "teacher_forcing": true,
55
+ "max_grad_norm": null,
56
+ "max_train_batches": null,
57
+ "max_eval_batches": null,
58
+ "weight_dir": "weight",
59
+ "checkpoint_name": "kno_navier_stokes.pt"
60
+ },
61
+ "inference": {
62
+ "result_dir": "result",
63
+ "num_samples": 4
64
+ }
65
+ }
66
+ }