Hugging Face's logo

OneScience-Group
/
MeshGraphNet

English
OneScience
fluid dynamics
external flow prediction
unstructured-mesh simulation
Model card Files
xet
 Community
MeshGraphNet
File size: 1,983 Bytes
4dfb91c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
{
  "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
  }
}