yushuang88 commited on
Commit
f602b9b
·
verified ·
1 Parent(s): c29de75

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +84 -0
config.json ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "FPINNForward",
4
+ "FPINNInverse"
5
+ ],
6
+ "framework": "pytorch",
7
+ "model_name": "FPINNs",
8
+ "model_type": "fpinns",
9
+ "source_config": "conf/config.yaml",
10
+ "root": {
11
+ "common": {
12
+ "task": "forward",
13
+ "device": "auto",
14
+ "dtype": "float32",
15
+ "seed": 1111,
16
+ "weight_dir": "weight",
17
+ "result_dir": "result"
18
+ },
19
+ "data": {
20
+ "mat_file": "data/AC.mat",
21
+ "n_train": 8000,
22
+ "evaluation_batch_size": 8192
23
+ },
24
+ "model": {
25
+ "hidden_layers": [
26
+ 200,
27
+ 200,
28
+ 200,
29
+ 200
30
+ ],
31
+ "linear_dim": 10,
32
+ "fuzzy_dim": 4,
33
+ "activation": "tanh"
34
+ },
35
+ "tasks": {
36
+ "forward": {
37
+ "training": {
38
+ "epochs": 10000,
39
+ "lr": 0.001,
40
+ "lbfgs_iters": 0,
41
+ "log_interval": 500
42
+ },
43
+ "loss": {
44
+ "data": 1.0,
45
+ "pde": 1.0
46
+ },
47
+ "pde": {
48
+ "lambda_1": 0.0001,
49
+ "lambda_2": 5.0
50
+ },
51
+ "output": {
52
+ "checkpoint_name": "fpinn_forward.pt",
53
+ "history_name": "fpinn_forward_history.npz",
54
+ "predictions_name": "fpinn_forward.npz",
55
+ "figure_name": "fpinn_forward.png"
56
+ }
57
+ },
58
+ "inverse": {
59
+ "training": {
60
+ "epochs": 10000,
61
+ "lr": 0.001,
62
+ "lbfgs_iters": 0,
63
+ "log_interval": 500
64
+ },
65
+ "loss": {
66
+ "data": 1.0,
67
+ "pde": 1.0
68
+ },
69
+ "pde": {
70
+ "initial_lambda_1": 1.0,
71
+ "initial_lambda_2": 0.0,
72
+ "true_lambda_1": 0.0001,
73
+ "true_lambda_2": 5.0
74
+ },
75
+ "output": {
76
+ "checkpoint_name": "fpinn_inverse.pt",
77
+ "history_name": "fpinn_inverse_history.npz",
78
+ "predictions_name": "fpinn_inverse.npz",
79
+ "figure_name": "fpinn_inverse.png"
80
+ }
81
+ }
82
+ }
83
+ }
84
+ }