File size: 1,399 Bytes
d46a58d | 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 | paths:
data_dir: data/cylinder_flow
fake_data_path: data/cylinder_flow/fake_cylinder_flow.pt
checkpoint_dir: weight/checkpoints
result_dir: results
datapipe:
source:
data_dir: data/cylinder_flow
stats_dir: data/cylinder_flow/stats
fake_data: true
fake_data_path: data/cylinder_flow/fake_cylinder_flow.pt
verbose: true
data:
train_samples: 4
train_steps: 5
val_samples: 2
val_steps: 5
test_samples: 2
test_steps: 5
noise_std: 0.0
dataloader:
batch_size: 1
num_workers: 0
model:
name: MeshGraphNet
specific_params:
MeshGraphNet:
num_input_features: 6
num_edge_features: 3
num_output_features: 3
processor_size: 2
hidden_dim_processor: 32
num_layers_node_processor: 2
num_layers_edge_processor: 2
hidden_dim_node_encoder: 32
hidden_dim_edge_encoder: 32
hidden_dim_node_decoder: 32
do_concat_trick: false
num_processor_checkpoint_segments: 0
recompute_activation: false
training:
device: cpu
max_epoch: 1
lr: 0.0001
lr_decay_rate: 0.999
amp: false
jit: false
checkpoint_dir: weight/checkpoints
loss_criterion: MSE
patience: 10
log_interval: 1
gpuid: 0
inference:
device: cpu
checkpoint_dir: weight/checkpoints
output_path: results/inference_results.npz
viz_vars: ["u", "v", "p"]
frame_skip: 1
frame_interval: 1
|