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

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +57 -0
config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ }