End of training

README.md

@@ -37,7 +37,7 @@ image.save("my_image.png")
 
 These are the key hyperparameters used during training:
 
-* Epochs: 50
+* Epochs: 750
 * Learning rate: 1e-05
 * Batch size: 16
 * Gradient accumulation steps: 4
@@ -45,4 +45,4 @@ These are the key hyperparameters used during training:
 * Mixed-precision: None
 
 
-More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/vishnou/text2image-fine-tune/runs/6yidwefq).
+More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/vishnou/text2image-fine-tune/runs/xwkl09i1).