File size: 779 Bytes
8880eca | 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 | root:
datapipe:
verbose: false
source:
data_dir: "./data/deepcfd"
data_x_name: "dataX.pkl"
data_y_name: "dataY.pkl"
data:
split_ratio: 0.75
seed: 0
dataloader:
batch_size: 1
num_workers: 0
model:
name: "UNetEx"
in_channels: 3
out_channels: 3
base_channels: 2
num_stages: 1
kernel_size: 3
bilinear: true
normtype: "none"
training:
output_dir: "./weight"
checkpoint_name: "best_model.pt"
num_epochs: 1
lr: 0.001
weight_decay: 0.005
patience: 20
eval_interval: 1
inference:
checkpoint_path: "./weight/best_model.pt"
result_dir: "./result/deepcfd"
num_visualize: 2
fake_data:
num_samples: 4
height: 16
width: 16
seed: 42
|