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

Fix config

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1 +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"}
 
1
+ {"architecture": "Perceiver", "input_channels": 16, "sat_channels": 12, "forecast_steps": 6, "lr": 0.002, "input_size": 32, "max_frequency": 16.0, "depth": 6, "num_latents": 256, "cross_heads": 1, "latent_heads": 8, "cross_dim_heads": 8, "latent_dim": 256, "weight_tie_layers": false, "self_per_cross_attention": 2, "dim": 32, "logits_dim": null, "queries_dim": 32, "latent_dim_heads": 64, "visualize": true, "preprocessor_type": "metnet"}