End of training

README.md

@@ -21,12 +21,12 @@ This pipeline was finetuned from **None** on the **uoft-cs/cifar10** dataset. Be
 
 These are the key hyperparameters used during training:
 
-* Epochs: 10
-* Learning rate: 0.0003072
-* Batch size: 1024
+* Epochs: 100
+* Learning rate: 0.0006144
+* Batch size: 2048
 * Gradient accumulation steps: 2
 * Image resolution: 28
 * Mixed-precision: fp16
 
 
-More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/wearesameasyou/vae-fine-tune/runs/m1s1jrio).
+More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/wearesameasyou/vae-fine-tune/runs/8ukoviyx).