config.json

{"input_channels": 17, "sat_channels": 13, "forecast_steps": 24, "input_size": 64, "lr": 0.0005, "visualize": true, "max_frequency": 4.0, "depth": 6, "num_latents": 256, "cross_heads": 1, "latent_heads": 8, "cross_dim_heads": 8, "latent_dim": 512, "weight_tie_layers": false, "decoder_ff": true, "dim": 32, "logits_dim": 100, "queries_dim": 32, "latent_dim_heads": 64, "loss": "mse", "sin_only": false, "encode_fourier": true, "preprocessor_type": null, "postprocessor_type": null, "encoder_kwargs": null, "decoder_kwargs": null, "pretrained": false}