yushuang88 commited on
Commit
69004da
·
verified ·
1 Parent(s): 10ab16f

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +56 -0
config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BPINN"
4
+ ],
5
+ "framework": "pytorch",
6
+ "model_name": "BPINNs",
7
+ "model_type": "bpinns",
8
+ "source_config": "conf/config.yaml",
9
+ "root": {
10
+ "common": {
11
+ "device": "auto",
12
+ "dtype": "float64",
13
+ "seed": 1234,
14
+ "weight_dir": "weight",
15
+ "result_dir": "result"
16
+ },
17
+ "model": {
18
+ "name": "BPINN",
19
+ "input_dim": 1,
20
+ "n_layers": 4,
21
+ "n_neurons": 50,
22
+ "n_out_sol": 1,
23
+ "n_out_par": 0,
24
+ "activation": "tanh"
25
+ },
26
+ "training": {
27
+ "epochs": 10000,
28
+ "lr": 0.001,
29
+ "lbfgs_iters": 2000,
30
+ "refinement_lr": 1.0,
31
+ "log_interval": 500,
32
+ "checkpoint_name": "bpinn_laplace1d.pt",
33
+ "refined_checkpoint_name": "bpinn_laplace1d_refined.pt"
34
+ },
35
+ "loss": {
36
+ "data": 10.0,
37
+ "pde": 1.0,
38
+ "boundary": 10.0
39
+ },
40
+ "data": {
41
+ "domain": [
42
+ 0.0,
43
+ 1.0
44
+ ],
45
+ "n_sol": 200,
46
+ "n_pde": 1000,
47
+ "noise_std": 0.0,
48
+ "test_res": 200
49
+ },
50
+ "inference": {
51
+ "predictions_name": "bpinn_laplace1d.npz",
52
+ "figure_name": "bpinn_laplace1d.png",
53
+ "history_name": "training_history.npz"
54
+ }
55
+ }
56
+ }