metnet-2 / config.json
jacobbieker's picture
Add model
c7566a2
{"image_encoder": "downsampler", "input_channels": 12, "input_size": 512, "lstm_channels": 128, "lead_time_features": 2048, "upsample_method": "interp", "num_upsampler_blocks": 2, "num_context_blocks": 3, "num_input_timesteps": 13, "encoder_dilations": [1, 2, 4, 8, 16, 32, 64, 128], "sat_channels": 12, "output_channels": 12, "center_crop_size": 128, "forecast_steps": 48, "use_preprocessor": true, "kwargs": {}, "encoder_channels": 384, "kernel_size": 3, "upsampler_channels": 512}