Upload models_shortcut/configs.yml with huggingface_hub

models_shortcut/configs.yml

@@ -0,0 +1,59 @@
+model_v0_mtg_meteofrance:
+  log_every_n_steps: 5
+  save_every_n_epochs: 5
+  model_dir: "models/"
+  parametrization: "endpoint"
+  batch_size: 128
+  learning_rate: 0.001
+  num_epochs: 200
+  seed: 42
+  gradient_clip_value: 1.0
+  dataset_path: "/workspace/dataset"
+
+model_v0_mtg_lightning:
+  log_every_n_steps: 5
+  save_every_n_epochs: 5
+  model_dir: "models/"
+  parametrization: "standard"
+  batch_size: 64
+  learning_rate: 0.001
+  num_epochs: 200
+  seed: 42
+  gradient_clip_value: 1.0
+  dataset_path: "/workspace/dataset"
+  model:
+    sat_in_channels: 12
+    kpi_in_channels: 1
+    sat_out_channels: 12
+    kpi_out_channels: 1
+    additional_channels: 3
+    features: [32, 64, 128, 256]
+    context_dim: 4
+    embedding_dim: 128
+    context_frames: 4
+    num_additional_resnet_blocks: 2
+    time_emb_dim: 64
+
+model_v0_mtg_lightning_shortcut:
+  log_every_n_steps: 5
+  save_every_n_epochs: 5
+  model_dir: "models/"
+  parametrization: "standard"
+  batch_size: 64
+  learning_rate: 0.001
+  num_epochs: 200
+  seed: 42
+  gradient_clip_value: 1.0
+  dataset_path: "/workspace/dataset"
+  model:
+    sat_in_channels: 12
+    kpi_in_channels: 1
+    sat_out_channels: 12
+    kpi_out_channels: 1
+    additional_channels: 3
+    features: [32, 64, 128, 256]
+    context_dim: 5
+    embedding_dim: 128
+    context_frames: 4
+    num_additional_resnet_blocks: 2
+    time_emb_dim: 64