Jacob Bieker commited on
Commit
07049ad
·
1 Parent(s): bfb11aa

Add config

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"input_channels": 16, "sat_channels": 12, "forecast_steps": 6, "lr": 0.005, "input_size": 32, "max_frequency": 16.0, "depth": 6, "num_latents": 256, "cross_heads": 1, "latent_heads": 8, "cross_dim_heads": 4, "latent_dim": 256, "weight_tie_layers": false, "self_per_cross_attention": 4, "dim": 32, "logits_dim": null, "queries_dim": 32, "latent_dim_heads": 64, "visualize": true, "preprocessor_type": "metnet"}