config.json

{
  "model_type": "crop_navier_stokes",
  "architectures": ["CROPNavierStokes"],
  "input_channels": 1,
  "output_channels": 1,
  "resolution": 64,
  "modes": 24,
  "initial_channels": 32,
  "levels": 3,
  "residual_blocks_per_level": 4,
  "bottleneck_blocks": 6,
  "kernel_size": 3,
  "latent_size": 64,
  "use_bn": true,
  "checkpoint": "weight/best_model.pth"
}