ZeroCool94
commited on
Commit
•
9701c55
1
Parent(s):
23594eb
Update README.md
Browse files
README.md
CHANGED
@@ -66,6 +66,8 @@ The model was trained on the following dataset:
|
|
66 |
- **Resolution/Image Size**: First trained at a resolution of 64x64, then increased to 256x256 and then to 512x512. Check the notes down below for more details on this.
|
67 |
- **Dimension:** 128
|
68 |
- **vq_codebook_size:** 8192
|
|
|
|
|
69 |
- **Total Training Steps:** 1,999,500
|
70 |
|
71 |
Note: On Muse we can change the image_size or resolution at any time without having to train the model from scratch again, this allows us to first train the model at low resolution using the same `dim` and `vq_codebook_size` to train faster and then we can increase the `image_size` and use a higher resolution once the model has trained enough.
|
|
|
66 |
- **Resolution/Image Size**: First trained at a resolution of 64x64, then increased to 256x256 and then to 512x512. Check the notes down below for more details on this.
|
67 |
- **Dimension:** 128
|
68 |
- **vq_codebook_size:** 8192
|
69 |
+
- **heads:** 8
|
70 |
+
- **depth:** 4
|
71 |
- **Total Training Steps:** 1,999,500
|
72 |
|
73 |
Note: On Muse we can change the image_size or resolution at any time without having to train the model from scratch again, this allows us to first train the model at low resolution using the same `dim` and `vq_codebook_size` to train faster and then we can increase the `image_size` and use a higher resolution once the model has trained enough.
|