Update README.md

README.md

@@ -61,5 +61,9 @@ The model was trained on the following dataset:
 - **Batch:** 1
 - **Learning Rate:** 7e-08
 - **Warmup Steps:** 10,000
-- **Resolution**: 512 pixels
+- **Resolution/Image Size**: First trained at a resolution of 64x64 and then increased to 384x384. Check the notes down below for more details on this.
+- **Dimension:** 256
+- **vq_codebook_size:** 256
 - **Total Training Steps:** 1,190,500
+
+  Note: On Muse we can change the image_size or resolution at any time without having to train the model from scratch again, this allow 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.