Jacob Bieker
commited on
Commit
•
f86f1fd
1
Parent(s):
07049ad
Fix config
Browse files- config.json +1 -1
config.json
CHANGED
@@ -1 +1 @@
|
|
1 |
-
{"input_channels": 16, "sat_channels": 12, "forecast_steps": 6, "lr": 0.
|
|
|
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"}
|