yushuang88 commited on
Commit
95c5863
·
verified ·
1 Parent(s): 69004da

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +7 -54
config.json CHANGED
@@ -1,56 +1,9 @@
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
  }
 
1
  {
2
+ "name": "BPINN",
3
+ "input_dim": 1,
4
+ "n_layers": 4,
5
+ "n_neurons": 50,
6
+ "n_out_sol": 1,
7
+ "n_out_par": 0,
8
+ "activation": "tanh"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  }