config.json

{
  "model_type": "neural_convection_compact_fno_2d",
  "architectures": ["NeuralConvectionCompactFNO2D"],
  "in_channels": 4,
  "out_channels": 4,
  "channel_names": ["u_x", "u_y", "T", "p"],
  "grid_size": [64, 64],
  "width": 64,
  "fourier_blocks": 4,
  "modes": [16, 16],
  "checkpoint": "weight/best_model.pth",
  "paper": "https://arxiv.org/abs/2606.25259",
  "dataset": "NurshatMenglik/NeuralConvection_2D_TrainTest",
  "dataset_revision": "v1.0-legacy"
}