yushuang88 commited on
Commit
ba1f5e5
·
verified ·
1 Parent(s): b065692

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +68 -0
config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "GPPLUS"
4
+ ],
5
+ "framework": "pytorch",
6
+ "model_name": "GP_for_TO",
7
+ "model_type": "gp_for_to",
8
+ "source_config": "conf/config.yaml",
9
+ "root": {
10
+ "problem": "doublepipe",
11
+ "seed": 11,
12
+ "output_names": [
13
+ "u",
14
+ "v",
15
+ "p",
16
+ "ro"
17
+ ],
18
+ "runtime": {
19
+ "onescience_src": null
20
+ },
21
+ "data": {
22
+ "n_train_per_bc": 25,
23
+ "n_col_domain": 10000
24
+ },
25
+ "model": {
26
+ "mean_function": "neural_network",
27
+ "nn_layers_base": [
28
+ 64,
29
+ 64,
30
+ 64,
31
+ 64,
32
+ 64
33
+ ],
34
+ "dtype": "float32"
35
+ },
36
+ "training": {
37
+ "device": "auto",
38
+ "gpu": 0,
39
+ "num_iter": 50000,
40
+ "lr_default": 0.001,
41
+ "diff_method": "Numerical",
42
+ "checkpoint_path": "./weight/gp_for_to.pt",
43
+ "output_dir": "./result/training",
44
+ "plot_outputs": true,
45
+ "checkpoints": [
46
+ 1000,
47
+ 10000,
48
+ 20000,
49
+ 30000,
50
+ 40000,
51
+ 50000
52
+ ]
53
+ },
54
+ "inference": {
55
+ "device": "auto",
56
+ "gpu": 0,
57
+ "checkpoint_path": "./weight/gp_for_to.pt",
58
+ "output_dir": "./result/inference",
59
+ "n_col_domain": 10000
60
+ },
61
+ "fake_data": {
62
+ "problem": "doublepipe",
63
+ "n_train_per_bc": 25,
64
+ "n_col_domain": 10000,
65
+ "output_dir": "./data/fake"
66
+ }
67
+ }
68
+ }