yushuang88 commited on
Commit
321bcd1
·
verified ·
1 Parent(s): eb6db06

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +29 -73
config.json CHANGED
@@ -1,79 +1,35 @@
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
  }
 
1
  {
2
+ "name": "XPINNPoisson2D",
3
+ "subnetworks": {
4
+ "domain1": {
5
+ "layers": [
6
+ 2,
7
+ 30,
8
+ 30,
9
+ 1
10
+ ],
11
+ "activation": "tanh"
 
 
 
 
12
  },
13
+ "domain2": {
14
+ "layers": [
15
+ 2,
16
+ 20,
17
+ 20,
18
+ 20,
19
+ 20,
20
+ 1
21
+ ],
22
+ "activation": "sin"
23
  },
24
+ "domain3": {
25
+ "layers": [
26
+ 2,
27
+ 25,
28
+ 25,
29
+ 25,
30
+ 1
31
+ ],
32
+ "activation": "cos"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  }
34
  }
35
  }