Upload models_world_shortcut/configs.yml with huggingface_hub

models_world_shortcut/configs.yml

@@ -91,11 +91,11 @@ model_v1_mtg_world_lightning_shortcut:
   model_dir: "models/"
   parametrization: "standard"
   interpolation: "polynomial"
-  batch_size: 64
+  batch_size: 128
   learning_rate: 0.001
   num_epochs: 200
   seed: 42
-  sigma_noise_input: 0.
+  sigma_noise_input: 0.01
   gradient_clip_value: 1.0
   dataset_path: "/workspace/dataset"
   model:
@@ -136,4 +136,103 @@ model_v2_mtg_world_lightning_shortcut:
     embedding_dim: 128
     context_frames: 4
     num_additional_resnet_blocks: 2
+    time_emb_dim: 64
+
+
+model_v3_mtg_world_lightning_shortcut:
+  log_every_n_steps: 5
+  save_every_n_epochs: 5
+  model_dir: "models/"
+  parametrization: "standard"
+  interpolation: "polynomial"
+  batch_size: 64
+  learning_rate: 0.001
+  num_epochs: 200
+  seed: 42
+  gen_steps: 4
+  sigma_noise_input: 0.01
+  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_correction:
+    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: 2
+    num_additional_resnet_blocks: 2
+    time_emb_dim: 64
+
+
+
+model_v4_mtg_world_lightning_shortcut:
+  log_every_n_steps: 5
+  save_every_n_epochs: 5
+  model_dir: "models/"
+  parametrization: "standard"
+  interpolation: "polynomial"
+  batch_size: 64
+  learning_rate: 0.001
+  num_epochs: 200
+  seed: 42
+  gen_steps: 4
+  sigma_noise_input: 0.01
+  gradient_clip_value: 1.0
+  residual: False
+  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_v5_mtg_world_lightning_shortcut:
+  log_every_n_steps: 5
+  save_every_n_epochs: 5
+  model_dir: "models/"
+  parametrization: "standard"
+  interpolation: "polynomial"
+  batch_size: 64
+  learning_rate: 0.001
+  num_epochs: 200
+  seed: 42
+  gen_steps: 4
+  sigma_noise_input: 0.005
+  gradient_clip_value: 1.0
+  residual: True
+  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: [64, 128, 256, 512]
+    context_dim: 5
+    embedding_dim: 128
+    context_frames: 4
+    num_additional_resnet_blocks: 2
     time_emb_dim: 64