| { |
| "architectures": [ |
| "MeshGraphNet" |
| ], |
| "framework": "pytorch", |
| "model_name": "MeshGraphNet", |
| "model_type": "meshgraphnet", |
| "source_config": "config/config.yaml", |
| "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 |
| } |
| } |
|
|
