Vishnou commited on
Commit
2ac2bb5
1 Parent(s): 72cfc83

End of training

Browse files
Files changed (2) hide show
  1. README.md +3 -3
  2. val_imgs_grid.png +0 -0
README.md CHANGED
@@ -37,12 +37,12 @@ image.save("my_image.png")
37
 
38
  These are the key hyperparameters used during training:
39
 
40
- * Epochs: 750
41
  * Learning rate: 1e-05
42
  * Batch size: 16
43
  * Gradient accumulation steps: 4
44
  * Image resolution: 512
45
- * Mixed-precision: None
46
 
47
 
48
- 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).
 
37
 
38
  These are the key hyperparameters used during training:
39
 
40
+ * Epochs: 1
41
  * Learning rate: 1e-05
42
  * Batch size: 16
43
  * Gradient accumulation steps: 4
44
  * Image resolution: 512
45
+ * Mixed-precision: bf16
46
 
47
 
48
+ 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/dnbasjwq).
val_imgs_grid.png CHANGED