File size: 1,168 Bytes
21e6c23 | 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 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | model:
name: "Transolver"
specific_params:
Transolver: &transolver_params
n_hidden: 32
n_layers: 2
space_dim: 7
fun_dim: 0
n_head: 4
mlp_ratio: 2
out_dim: 4
slice_num: 8
unified_pos: 0
build_graph: false
r: 0.05
max_neighbors: 32
Transolver_plus: *transolver_params
datapipe:
name: "AirfRANS"
backend: "fake_airfrans"
verbose: false
source:
data_dir: "./data/fake_airfrans"
stats_dir: "./data/stats"
data:
splits:
task: "fake"
train_name: "train"
val_name: "val"
test_name: "test"
val_split_ratio: 0.25
sampling:
sample_strategy: null
n_boot: 128
surf_ratio: 0.25
crop: null
subsampling: 64
dataloader:
batch_size: 1
num_workers: 0
training:
device: "cpu"
max_epoch: 1
lr: 0.001
gpuid: 0
patience: 20
progress_log_interval: 5
loss_criterion: "MSE_weighted"
loss_weight: 1.0
checkpoint_dir: "./weight"
result_dir: "./result"
n_test: 2
seed: 42
fake_data:
num_train: 4
num_val: 2
num_test: 2
num_nodes: 64
input_dim: 7
output_dim: 4
surface_ratio: 0.25
|