Upload config.json with huggingface_hub

config.json

@@ -1,5 +1,5 @@
 {
-  "save_path": "saved_bsz64_acc1_lr8e5_512_leftclick_histpos_512_384_cont2_ddd",
+  "save_path": "saved_standard_challenging_context32_nocond_cont_cont_all_cont",
   "model": {
     "base_learning_rate": 8e-05,
     "target": "latent_diffusion.ldm.models.diffusion.ddpm.LatentDiffusion",
@@ -18,7 +18,7 @@
         48
       ],
       "channels": 3,
-      "cond_stage_trainable": true,
+      "cond_stage_trainable": false,
       "conditioning_key": "hybrid",
       "monitor": "val/loss_simple_ema",
       "unet_config": {
@@ -28,7 +28,7 @@
             64,
             48
           ],
-          "in_channels": 41,
+          "in_channels": 166,
           "out_channels": 4,
           "model_channels": 192,
           "attention_resolutions": [
@@ -44,9 +44,8 @@
             5
           ],
           "num_head_channels": 32,
-          "use_spatial_transformer": true,
-          "transformer_depth": 1,
-          "context_dim": 768
+          "use_spatial_transformer": false,
+          "transformer_depth": 1
         }
       },
       "first_stage_config": {
@@ -76,20 +75,14 @@
           }
         }
       },
-      "cond_stage_config": {
-        "target": "latent_diffusion.ldm.modules.encoders.modules.GPTEmbedder",
-        "params": {
-          "n_embed": 768,
-          "n_layer": 12
-        }
-      }
+      "cond_stage_config": "__is_unconditional__"
     }
   },
   "data": {
     "target": "data.data_processing.datasets.DataModule",
     "params": {
-      "batch_size": 8,
-      "num_workers": 4,
+      "batch_size": 256,
+      "num_workers": 16,
       "wrap": false,
       "shuffle": true,
       "drop_last": true,
@@ -99,7 +92,9 @@
       "train": {
         "target": "data.data_processing.datasets.ActionsData",
         "params": {
-          "data_csv_path": "train_dataset/train_dataset_20000.csv"
+          "data_csv_path": "train_dataset/train_dataset.target_frames.csv",
+          "normalization": "standard",
+          "context_length": 32
         }
       }
     }
@@ -107,7 +102,7 @@
   "lightning": {
     "trainer": {
       "benchmark": false,
-      "max_epochs": 64,
+      "max_epochs": 6400,
       "limit_val_batches": 0,
       "accelerator": "gpu",
       "gpus": 1,