Update README.md

README.md

@@ -72,4 +72,24 @@ images = pipe(
 images[0].save(f"hug_lab.png")
 ```
 
-![images_10)](./out_hug_lab_7.png)
+![images_10)](./out_hug_lab_7.png)
+
+### Training
+
+#### Training data
+This checkpoint was first trained for 20,000 steps on laion 6a resized to a max minimum dimension of 384. 
+It was then further trained for 20,000 steps on laion 6a resized to a max minimum dimension of 1024 and 
+then filtered to contain only minimum 1024 images. We found the further high resolution finetuning was 
+necessary for image quality.
+
+#### Compute
+one 8xA100 machine
+
+#### Batch size
+Data parallel with a single gpu batch size of 8 for a total batch size of 64.
+
+#### Hyper Parameters
+Constant learning rate of 1e-4 scaled by batch size for total learning rate of 64e-4
+
+#### Mixed precision
+fp16