File size: 1,539 Bytes
338c3e4 | 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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | root:
datapipe:
source:
data_dir: "./data/cfdbench"
data_name: "tube_prop_bc_geo"
verbose: true
data:
task_type: "auto"
norm_props: true
norm_bc: true
delta_time: 0.1
stable_state_diff: 0.001
split_ratios: [0.7, 0.15, 0.15]
seed: 0
num_rows: 8
num_cols: 8
dataloader:
batch_size: 2
eval_batch_size: 2
num_workers: 0
model:
name: "fno"
num_label_samples: 64
in_chan: 2
out_chan: 2
ffn_depth: 2
ffn_width: 64
autoffn_depth: 2
autoffn_width: 64
deeponet_width: 64
branch_depth: 2
trunk_depth: 2
act_fn: "relu"
act_scale_invariant: true
act_on_output: false
autoedeeponet_width: 64
autoedeeponet_depth: 2
autoedeeponet_act_fn: "relu"
fno_depth: 2
fno_hidden_dim: 16
fno_modes_x: 4
fno_modes_y: 4
unet_dim: 8
unet_insert_case_params_at: "input"
resnet_depth: 2
resnet_hidden_chan: 16
resnet_kernel_size: 7
resnet_padding: 3
training:
output_dir: "./result"
mode: "train_test"
group_by_model: true
device: "cpu"
num_epochs: 1
lr: 1.0e-3
lr_step_size: 20
lr_gamma: 0.9
log_interval: 1
eval_interval: 1
loss_name: "nmse"
checkpoint_name: "auto"
inference:
checkpoint_path: "auto"
output_dir: "./result/inference"
group_by_model: true
device: "cpu"
max_batches: 2
plot_interval: 1
fake_data:
num_cases_per_subset: 4
num_frames: 4
seed: 123
|