Upload decoder/2B/decoder_config.json with huggingface_hub

decoder/2B/decoder_config.json

@@ -30,7 +30,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": 8,
+        "batch_size": 1,
         "start_shard": 0,
         "end_shard": 5247,
         "shard_width": 5,
@@ -71,7 +71,7 @@
         "find_unused_parameters": true
     },
     "evaluate": {
-        "n_evaluation_samples": 3,
+        "n_evaluation_samples": 10,
         "FID": {
             "feature": 64
         },
@@ -85,7 +85,7 @@
         "overwrite_data_path": true,
 
         "log": {
-            "log_type": "wandb",
+            "log_type": "console",
 
             "wandb_entity": "Veldrovive",
             "wandb_project": "dalle2_train_decoder",
@@ -102,21 +102,6 @@
         },
 
         "save": [{
-            "save_to": "huggingface",
-            "huggingface_repo": "laion/DALLE2-PyTorch",
-            
-            "save_meta_to": "decoder/2B/",
-            "save_latest_to": "decoder/2B/latest.pth",
-
-            "save_type": "model"
-        },{
-            "save_to": "huggingface",
-            "huggingface_repo": "laion/DALLE2-PyTorch",
-            
-            "save_latest_to": "decoder/2B/checkpoints/latest.pth",
-
-            "save_type": "checkpoint"
-        },{
             "save_to": "local",
             "save_latest_to": "/fsx/aidan/new/multinode/experiments/deepspeed/models/checkpoints/latest.pth",