yushuang88 commited on
Commit
5f6ff7a
·
verified ·
1 Parent(s): 7abe440

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +70 -0
config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "FNO1d",
4
+ "FNO2d",
5
+ "FNO3d"
6
+ ],
7
+ "framework": "pytorch",
8
+ "model_name": "PDENNEval",
9
+ "model_type": "pdenneval",
10
+ "source_config": "conf/config.yaml",
11
+ "fno_config": {
12
+ "datapipe": {
13
+ "source": {
14
+ "data_dir": "./data",
15
+ "file_name": "2D_DarcyFlow_beta0.1_Train.hdf5"
16
+ },
17
+ "verbose": true,
18
+ "data": {
19
+ "single_file": true,
20
+ "reduced_resolution": 1,
21
+ "reduced_resolution_t": 1,
22
+ "reduced_batch": 1,
23
+ "initial_step": 1,
24
+ "test_ratio": 0.1,
25
+ "pde_name": "2D_Darcy_Flow"
26
+ },
27
+ "dataloader": {
28
+ "batch_size": 2,
29
+ "num_workers": 0,
30
+ "pin_memory": false
31
+ }
32
+ },
33
+ "model": {
34
+ "name": "FNO",
35
+ "num_channels": 1,
36
+ "modes": 4,
37
+ "width": 8
38
+ },
39
+ "training": {
40
+ "output_dir": "./weight",
41
+ "if_training": true,
42
+ "continue_training": false,
43
+ "model_path": null,
44
+ "save_period": 1,
45
+ "seed": 0,
46
+ "training_type": "single",
47
+ "t_train": 2,
48
+ "epochs": 1,
49
+ "optimizer": {
50
+ "name": "Adam",
51
+ "lr": 0.001,
52
+ "weight_decay": 0.0001
53
+ },
54
+ "scheduler": {
55
+ "name": "StepLR",
56
+ "step_size": 100,
57
+ "gamma": 0.5
58
+ }
59
+ },
60
+ "inference": {
61
+ "checkpoint": "./weight/best_model.pt",
62
+ "output_dir": "./result/output",
63
+ "max_batches": 2
64
+ },
65
+ "result": {
66
+ "prediction_dir": "./result/output",
67
+ "output_dir": "./result"
68
+ }
69
+ }
70
+ }