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

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +4 -68
config.json CHANGED
@@ -1,70 +1,6 @@
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
  }
 
1
  {
2
+ "name": "FNO",
3
+ "num_channels": 1,
4
+ "modes": 4,
5
+ "width": 8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  }