| 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 | |