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: &params
      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