yushuang88 commited on
Commit
b9c4891
·
verified ·
1 Parent(s): 0061522

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +6 -63
config.json CHANGED
@@ -1,65 +1,8 @@
1
  {
2
- "architectures": [
3
- "PINNsformer1D"
4
- ],
5
- "framework": "pytorch",
6
- "model_name": "PINNsformer",
7
- "model_type": "pinnsformer",
8
- "source_config": "conf/config.yaml",
9
- "project": {
10
- "name": "PINNsformer",
11
- "task": "1d_reaction"
12
- },
13
- "runtime": {
14
- "seed": 0,
15
- "device": "auto"
16
- },
17
- "model": {
18
- "name": "PINNsformer1D",
19
- "d_out": 1,
20
- "d_hidden": 128,
21
- "d_model": 32,
22
- "num_layers": 1,
23
- "heads": 2
24
- },
25
- "data": {
26
- "x_range": [
27
- 0.0,
28
- 6.283185307179586
29
- ],
30
- "t_range": [
31
- 0.0,
32
- 1.0
33
- ],
34
- "x_num": 31,
35
- "t_num": 31,
36
- "sequence": {
37
- "num_step": 5,
38
- "step": 0.0001
39
- }
40
- },
41
- "equation": {
42
- "reaction_rate": 5.0,
43
- "initial": {
44
- "center": 3.141592653589793,
45
- "sigma": 0.7853981633974483
46
- }
47
- },
48
- "training": {
49
- "epochs": 2,
50
- "optimizer": {
51
- "name": "LBFGS",
52
- "lr": 1.0,
53
- "max_iter": 5,
54
- "line_search_fn": "strong_wolfe"
55
- },
56
- "checkpoint": "weight/1dreaction_pinnsformer.pt"
57
- },
58
- "paths": {
59
- "fake_data": "data/reaction_fake.npz",
60
- "prediction": "result/prediction.npz",
61
- "metrics": "result/metrics.json",
62
- "loss": "result/train_loss.npy",
63
- "figure": "result/1dreaction_pinnsformer.png"
64
- }
65
  }
 
1
  {
2
+ "name": "PINNsformer1D",
3
+ "d_out": 1,
4
+ "d_hidden": 128,
5
+ "d_model": 32,
6
+ "num_layers": 1,
7
+ "heads": 2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  }