File size: 929 Bytes
e9b4f6f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | model:
name: "Transolver_plus"
specific_params:
Transolver: ¶ms
n_hidden: 16
n_layers: 1
space_dim: 7
fun_dim: 0
n_head: 2
mlp_ratio: 2
out_dim: 4
slice_num: 4
unified_pos: 0
cfd_mesh: true
r: 0.2
Transolver_plus: *params
datapipe:
name: "ShapeNetCar"
task: "Transolver-Car-Design"
verbose: true
source:
data_dir: "./data/mlcfd_data/training_data"
preprocessed_save_dir: "./data/mlcfd_data/preprocessed_data"
stats_dir: "./data/mlcfd_data/stats"
preprocessed: 1
data:
splits:
fold_id: 0
dataloader:
batch_size: 1
num_workers: 0
pin_memory: false
training:
gpuid: -1
max_epoch: 1
patience: 50
lr: 0.001
loss_criterion: "MSE"
loss_weight: 0.5
checkpoint_dir: "./weight"
val_iter: 1
inference:
gpuid: -1
result_dir: "./results/ShapeNetCar"
save_vtk: false
visualize: false
|