Hugging Face's logo

OneScience-Group
/
Transolver-Airfoil-Design

English
OneScience
fluid dynamics
airfoil aerodynamic prediction
Model card Files
xet
 Community
yushuang88 commited on
Commit
e9a5c29
·
verified ·
1 Parent(s): d8aee92

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +28 -90
config.json CHANGED
@@ -1,95 +1,33 @@
1
  {
2
- "architectures": [
3
- "Transolver2D",
4
- "Transolver2D_plus"
5
- ],
6
- "framework": "pytorch",
7
- "model_name": "Transolver-Airfoil-Design",
8
- "model_type": "transolver_airfoil_design",
9
- "source_config": "conf/config.yaml",
10
- "model": {
11
- "name": "Transolver",
12
- "specific_params": {
13
- "Transolver": {
14
- "n_hidden": 32,
15
- "n_layers": 2,
16
- "space_dim": 7,
17
- "fun_dim": 0,
18
- "n_head": 4,
19
- "mlp_ratio": 2,
20
- "out_dim": 4,
21
- "slice_num": 8,
22
- "unified_pos": 0,
23
- "build_graph": false,
24
- "r": 0.05,
25
- "max_neighbors": 32
26
- },
27
- "Transolver_plus": {
28
- "n_hidden": 32,
29
- "n_layers": 2,
30
- "space_dim": 7,
31
- "fun_dim": 0,
32
- "n_head": 4,
33
- "mlp_ratio": 2,
34
- "out_dim": 4,
35
- "slice_num": 8,
36
- "unified_pos": 0,
37
- "build_graph": false,
38
- "r": 0.05,
39
- "max_neighbors": 32
40
- }
41
- }
42
- },
43
- "datapipe": {
44
- "name": "AirfRANS",
45
- "backend": "fake_airfrans",
46
- "verbose": false,
47
- "source": {
48
- "data_dir": "./data/fake_airfrans",
49
- "stats_dir": "./data/stats"
50
- },
51
- "data": {
52
- "splits": {
53
- "task": "fake",
54
- "train_name": "train",
55
- "val_name": "val",
56
- "test_name": "test",
57
- "val_split_ratio": 0.25
58
- },
59
- "sampling": {
60
- "sample_strategy": null,
61
- "n_boot": 128,
62
- "surf_ratio": 0.25
63
- },
64
- "crop": null,
65
- "subsampling": 64
66
  },
67
- "dataloader": {
68
- "batch_size": 1,
69
- "num_workers": 0
 
 
 
 
 
 
 
 
 
 
70
  }
71
- },
72
- "training": {
73
- "device": "cpu",
74
- "max_epoch": 1,
75
- "lr": 0.001,
76
- "gpuid": 0,
77
- "patience": 20,
78
- "progress_log_interval": 5,
79
- "loss_criterion": "MSE_weighted",
80
- "loss_weight": 1.0,
81
- "checkpoint_dir": "./weight",
82
- "result_dir": "./result",
83
- "n_test": 2,
84
- "seed": 42
85
- },
86
- "fake_data": {
87
- "num_train": 4,
88
- "num_val": 2,
89
- "num_test": 2,
90
- "num_nodes": 64,
91
- "input_dim": 7,
92
- "output_dim": 4,
93
- "surface_ratio": 0.25
94
  }
95
  }
 
1
  {
2
+ "name": "Transolver",
3
+ "specific_params": {
4
+ "Transolver": {
5
+ "n_hidden": 32,
6
+ "n_layers": 2,
7
+ "space_dim": 7,
8
+ "fun_dim": 0,
9
+ "n_head": 4,
10
+ "mlp_ratio": 2,
11
+ "out_dim": 4,
12
+ "slice_num": 8,
13
+ "unified_pos": 0,
14
+ "build_graph": false,
15
+ "r": 0.05,
16
+ "max_neighbors": 32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  },
18
+ "Transolver_plus": {
19
+ "n_hidden": 32,
20
+ "n_layers": 2,
21
+ "space_dim": 7,
22
+ "fun_dim": 0,
23
+ "n_head": 4,
24
+ "mlp_ratio": 2,
25
+ "out_dim": 4,
26
+ "slice_num": 8,
27
+ "unified_pos": 0,
28
+ "build_graph": false,
29
+ "r": 0.05,
30
+ "max_neighbors": 32
31
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
  }
33
  }