Upload decoder_config.json with huggingface_hub

decoder_config.json

@@ -27,7 +27,7 @@
                 "dim_mults": [1, 2, 3, 4],
                 "num_resnet_blocks": 3,
                 "sparse_attn": false,
-                "memory_efficient": true,
+                "memory_efficient": false,
                 "self_attn": [false, false, false, false]
             }
         ],
@@ -42,7 +42,7 @@
         "webdataset_base_url": "pipe:aws s3 cp --quiet s3://s-datasets/laion-aesthetic/data/laion2B-en-aesthetic/{}.tar -", 
         "img_embeddings_url": "s3://s-datasets/laion-aesthetic/ordered_embeddings/",
         "num_workers": 6,
-        "batch_size": 7,
+        "batch_size": 3,
         "start_shard": 0,
         "end_shard": 5247,
         "shard_width": 5,
@@ -68,11 +68,11 @@
         "lr": 1.2e-4,
         "wd": 0.0,
         "max_grad_norm": 0.5,
-        "save_every_n_samples": 1000000,
+        "save_every_n_samples": 400000,
         "n_sample_images": 10,
         "device": "cuda:0",
-        "epoch_samples": 5500000,
-        "validation_samples": 100000,
+        "epoch_samples": 2000000,
+        "validation_samples": 60000,
         "use_ema": true,
         "ema_beta": 0.9999,
         "save_all": false,