Hugging Face's logo

OneScience-Group
/
Transolver-Car-Design

English
OneScience
fluid dynamics
automotive aerodynamic design
CFD surrogate modeling
Model card Files
xet
 Community
yushuang88 commited on
Commit
5787d37
·
verified ·
1 Parent(s): 92ab382

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +26 -73
config.json CHANGED
@@ -1,78 +1,31 @@
1
  {
2
- "architectures": [
3
- "Transolver3D",
4
- "Transolver3D_plus"
5
- ],
6
- "framework": "pytorch",
7
- "model_name": "Transolver-Car-Design",
8
- "model_type": "transolver_car_design",
9
- "source_config": "conf/config.yaml",
10
- "model": {
11
- "name": "Transolver_plus",
12
- "specific_params": {
13
- "Transolver": {
14
- "n_hidden": 16,
15
- "n_layers": 1,
16
- "space_dim": 7,
17
- "fun_dim": 0,
18
- "n_head": 2,
19
- "mlp_ratio": 2,
20
- "out_dim": 4,
21
- "slice_num": 4,
22
- "unified_pos": 0,
23
- "cfd_mesh": true,
24
- "r": 0.2
25
- },
26
- "Transolver_plus": {
27
- "n_hidden": 16,
28
- "n_layers": 1,
29
- "space_dim": 7,
30
- "fun_dim": 0,
31
- "n_head": 2,
32
- "mlp_ratio": 2,
33
- "out_dim": 4,
34
- "slice_num": 4,
35
- "unified_pos": 0,
36
- "cfd_mesh": true,
37
- "r": 0.2
38
- }
39
- }
40
- },
41
- "datapipe": {
42
- "name": "ShapeNetCar",
43
- "task": "Transolver-Car-Design",
44
- "verbose": true,
45
- "source": {
46
- "data_dir": "./data/mlcfd_data/training_data",
47
- "preprocessed_save_dir": "./data/mlcfd_data/preprocessed_data",
48
- "stats_dir": "./data/mlcfd_data/stats",
49
- "preprocessed": 1
50
- },
51
- "data": {
52
- "splits": {
53
- "fold_id": 0
54
- }
55
  },
56
- "dataloader": {
57
- "batch_size": 1,
58
- "num_workers": 0,
59
- "pin_memory": false
 
 
 
 
 
 
 
 
60
  }
61
- },
62
- "training": {
63
- "gpuid": -1,
64
- "max_epoch": 1,
65
- "patience": 50,
66
- "lr": 0.001,
67
- "loss_criterion": "MSE",
68
- "loss_weight": 0.5,
69
- "checkpoint_dir": "./weight",
70
- "val_iter": 1
71
- },
72
- "inference": {
73
- "gpuid": -1,
74
- "result_dir": "./results/ShapeNetCar",
75
- "save_vtk": false,
76
- "visualize": false
77
  }
78
  }
 
1
  {
2
+ "name": "Transolver_plus",
3
+ "specific_params": {
4
+ "Transolver": {
5
+ "n_hidden": 16,
6
+ "n_layers": 1,
7
+ "space_dim": 7,
8
+ "fun_dim": 0,
9
+ "n_head": 2,
10
+ "mlp_ratio": 2,
11
+ "out_dim": 4,
12
+ "slice_num": 4,
13
+ "unified_pos": 0,
14
+ "cfd_mesh": true,
15
+ "r": 0.2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  },
17
+ "Transolver_plus": {
18
+ "n_hidden": 16,
19
+ "n_layers": 1,
20
+ "space_dim": 7,
21
+ "fun_dim": 0,
22
+ "n_head": 2,
23
+ "mlp_ratio": 2,
24
+ "out_dim": 4,
25
+ "slice_num": 4,
26
+ "unified_pos": 0,
27
+ "cfd_mesh": true,
28
+ "r": 0.2
29
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  }
31
  }