Upload 2 files

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fe576c3f1683983a29ff2c186d0bbaad7bf7cde6175c3fec48c2c28f2c6ec6b6
-size 143025308
+oid sha256:2cd29c934ebe2d77a3fb3ae1c5d28acf158bd5d1d6dad2c47fa2e5e38c3ca90d
+size 143024980

training_configs.json

@@ -0,0 +1,21 @@
+{
+  "max_epoch": 50,
+  "dataset_name": "uoft-cs/cifar10",
+  "model_repo": "google/ddpm-cifar10-32",
+  "image_size": 32,
+  "image_field": "img",
+  "lr": 2e-05,
+  "lr_warmup_steps": 500,
+  "weight_decay": 1e-05,
+  "train_batch_size": 64,
+  "eval_batch_size": 64,
+  "gradient_accumulation_steps": 2,
+  "remote_repo_id": "FriedParrot/ddpm-cifar10-diffusion",
+  "checkpoint_save_dir": "./checkpoints",
+  "validation_save_dir": "./validation_samples",
+  "checkpoint_epoch": 5,
+  "overwrite_output_dir": true,
+  "reverse_diffusion_steps": 100,
+  "config_save_dir": "./training_configs.json",
+  "gradient_clipping": 1.0
+}