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