{
  "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
  }
}