Upload models_world_shortcut/configs.yml with huggingface_hub

models_world_shortcut/configs.yml

@@ -69,6 +69,7 @@ model_v0_mtg_world_lightning_shortcut:
   learning_rate: 0.001
   num_epochs: 200
   seed: 42
+  sigma_noise_input: 0.
   gradient_clip_value: 1.0
   dataset_path: "/workspace/dataset"
   model:
@@ -94,6 +95,34 @@ model_v1_mtg_world_lightning_shortcut:
   learning_rate: 0.001
   num_epochs: 200
   seed: 42
+  sigma_noise_input: 0.
+  gradient_clip_value: 1.0
+  dataset_path: "/workspace/dataset"
+  model:
+    sat_in_channels: 16
+    kpi_in_channels: 1
+    sat_out_channels: 16
+    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
+
+
+model_v2_mtg_world_lightning_shortcut:
+  log_every_n_steps: 5
+  save_every_n_epochs: 5
+  model_dir: "models/"
+  parametrization: "standard"
+  interpolation: "linear"
+  batch_size: 128
+  learning_rate: 0.001
+  num_epochs: 200
+  seed: 42
+  sigma_noise_input: 0.01
   gradient_clip_value: 1.0
   dataset_path: "/workspace/dataset"
   model: