yushuang88 commited on
Commit
6c4c05f
·
verified ·
1 Parent(s): edfa5a7

Add model configuration

Browse files
Files changed (1) hide show
  1. config.json +66 -0
config.json ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Model"
4
+ ],
5
+ "framework": "pytorch",
6
+ "model_name": "CFD_Benchmark",
7
+ "model_type": "cfd_benchmark",
8
+ "source_config": "config/config.yaml",
9
+ "project": {
10
+ "name": "CFD_Benchmark",
11
+ "package": "cfd_benchmark"
12
+ },
13
+ "runtime": {
14
+ "device": "cpu"
15
+ },
16
+ "paths": {
17
+ "data_dir": "./data/fake_airfoil",
18
+ "checkpoint_dir": "./checkpoints",
19
+ "result_dir": "./results",
20
+ "weight_path": "./checkpoints/transolver_fake.pt"
21
+ },
22
+ "data": {
23
+ "loader": "airfoil",
24
+ "task": "steady",
25
+ "ntrain": 2,
26
+ "ntest": 1,
27
+ "batch_size": 1,
28
+ "downsamplex": 20,
29
+ "downsampley": 10,
30
+ "normalize": false,
31
+ "norm_type": "UnitTransformer",
32
+ "space_dim": 2,
33
+ "fun_dim": 0,
34
+ "out_dim": 1,
35
+ "geotype": "unstructured"
36
+ },
37
+ "model": {
38
+ "name": "Transolver",
39
+ "n_hidden": 16,
40
+ "n_layers": 1,
41
+ "n_heads": 2,
42
+ "mlp_ratio": 1,
43
+ "dropout": 0.0,
44
+ "act": "gelu",
45
+ "slice_num": 4,
46
+ "unified_pos": 0,
47
+ "ref": 4
48
+ },
49
+ "train": {
50
+ "epochs": 1,
51
+ "lr": 0.001,
52
+ "weight_decay": 1e-05,
53
+ "optimizer": "AdamW",
54
+ "scheduler": "CosineAnnealingLR",
55
+ "pct_start": 0.3,
56
+ "max_grad_norm": null,
57
+ "derivloss": false,
58
+ "use_checkpoint": false,
59
+ "checkpoint_layers": "",
60
+ "save_name": "transolver_fake"
61
+ },
62
+ "inference": {
63
+ "eval": 1,
64
+ "vis_num": 0
65
+ }
66
+ }