yushuang88 commited on
Commit
b491a8a
·
verified ·
1 Parent(s): 8196090

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +8 -55
config.json CHANGED
@@ -1,57 +1,10 @@
1
  {
2
- "architectures": [
3
- "UNetEx"
4
- ],
5
- "framework": "pytorch",
6
- "model_name": "DeepCFD",
7
- "model_type": "deepcfd",
8
- "source_config": "config/config.yaml",
9
- "root": {
10
- "datapipe": {
11
- "verbose": false,
12
- "source": {
13
- "data_dir": "./data/deepcfd",
14
- "data_x_name": "dataX.pkl",
15
- "data_y_name": "dataY.pkl"
16
- },
17
- "data": {
18
- "split_ratio": 0.75,
19
- "seed": 0
20
- },
21
- "dataloader": {
22
- "batch_size": 1,
23
- "num_workers": 0
24
- }
25
- },
26
- "model": {
27
- "name": "UNetEx",
28
- "in_channels": 3,
29
- "out_channels": 3,
30
- "base_channels": 2,
31
- "num_stages": 1,
32
- "kernel_size": 3,
33
- "bilinear": true,
34
- "normtype": "none"
35
- },
36
- "training": {
37
- "output_dir": "./weight",
38
- "checkpoint_name": "best_model.pt",
39
- "num_epochs": 1,
40
- "lr": 0.001,
41
- "weight_decay": 0.005,
42
- "patience": 20,
43
- "eval_interval": 1
44
- },
45
- "inference": {
46
- "checkpoint_path": "./weight/best_model.pt",
47
- "result_dir": "./result/deepcfd",
48
- "num_visualize": 2
49
- },
50
- "fake_data": {
51
- "num_samples": 4,
52
- "height": 16,
53
- "width": 16,
54
- "seed": 42
55
- }
56
- }
57
  }
 
1
  {
2
+ "name": "UNetEx",
3
+ "in_channels": 3,
4
+ "out_channels": 3,
5
+ "base_channels": 2,
6
+ "num_stages": 1,
7
+ "kernel_size": 3,
8
+ "bilinear": true,
9
+ "normtype": "none"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  }