Upload config.json

config.json

@@ -19,7 +19,7 @@
         "channels": 3,
         "timesteps": 1000,
         "loss_type": "l2",
-        "beta_schedule": "cosine",
+        "beta_schedule": ["cosine"],
         "learned_variance": true
     },
     "data": {
@@ -49,7 +49,7 @@
     },
     "train": {
         "epochs": 40,
-        "lr":1e-4,
+        "lr":5e-5,
         "wd": 0.01,
         "max_grad_norm": 0.5,
         "save_every_n_samples": 200000,
@@ -59,6 +59,7 @@
         "validation_samples": 100000,
         "use_ema": true,
         "ema_beta": 0.99,
+        "ema_param_or_buffer_names_no_ema": ["_dummy"],
         "save_all": false,
         "save_latest": true,
         "save_best": true,
@@ -86,8 +87,8 @@
         "source": "wandb",
 
         "run_path": "nousr_laion/dalle2_train_decoder/3rrsm5gy",
-        "file_path": "latest.pth",
+        "file_path": "best.pth",
 
-        "resume": true
+        "resume": false
     }
 }