End of training

README.md

@@ -3,7 +3,7 @@
 license: creativeml-openrail-m
 base_model: None
 datasets:
-- uoft-cs/cifar10
+- vipseg
 tags:
 - stable-diffusion
 - stable-diffusion-diffusers
@@ -14,7 +14,7 @@ inference: true
     
 # Text-to-image finetuning - mobled37/vae-model-finetuned
 
-This pipeline was finetuned from **None** on the **uoft-cs/cifar10** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: Nothing: 
+This pipeline was finetuned from **None** on the **vipseg** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: Nothing: 
 
 
 ## Training info
@@ -22,11 +22,11 @@ This pipeline was finetuned from **None** on the **uoft-cs/cifar10** dataset. Be
 These are the key hyperparameters used during training:
 
 * Epochs: 100
-* Learning rate: 0.0006144
-* Batch size: 2048
+* Learning rate: 4.8e-06
+* Batch size: 16
 * Gradient accumulation steps: 2
-* Image resolution: 28
+* Image resolution: 30
 * 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/8ukoviyx).
+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/bgysrmyj).