yushuang88 commited on
Commit
eb6db06
·
verified ·
1 Parent(s): 9a6baa4

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +79 -0
config.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "XPINNPoisson2D"
4
+ ],
5
+ "framework": "pytorch",
6
+ "model_name": "XPINNs",
7
+ "model_type": "xpinns",
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
+ "data": {
18
+ "mat_file": "data/XPINN_2D_PoissonEqn.mat",
19
+ "samples": {
20
+ "residual_1": 5000,
21
+ "residual_2": 1800,
22
+ "residual_3": 1200,
23
+ "boundary": 200,
24
+ "interface_1": 100,
25
+ "interface_2": 100
26
+ }
27
+ },
28
+ "model": {
29
+ "name": "XPINNPoisson2D",
30
+ "subnetworks": {
31
+ "domain1": {
32
+ "layers": [
33
+ 2,
34
+ 30,
35
+ 30,
36
+ 1
37
+ ],
38
+ "activation": "tanh"
39
+ },
40
+ "domain2": {
41
+ "layers": [
42
+ 2,
43
+ 20,
44
+ 20,
45
+ 20,
46
+ 20,
47
+ 1
48
+ ],
49
+ "activation": "sin"
50
+ },
51
+ "domain3": {
52
+ "layers": [
53
+ 2,
54
+ 25,
55
+ 25,
56
+ 25,
57
+ 1
58
+ ],
59
+ "activation": "cos"
60
+ }
61
+ }
62
+ },
63
+ "loss": {
64
+ "boundary": 20.0,
65
+ "pde": 1.0,
66
+ "interface_residual": 1.0,
67
+ "interface_value": 20.0
68
+ },
69
+ "training": {
70
+ "steps": 501,
71
+ "lr": 0.0008,
72
+ "log_interval": 20,
73
+ "checkpoint_name": "xpinn_poisson_2d.pt"
74
+ },
75
+ "inference": {
76
+ "figure_name": "xpinn_poisson_2d.png"
77
+ }
78
+ }
79
+ }