Update README.md

README.md

@@ -91,22 +91,13 @@ Our training script was built on top of the official training script that we pro
 You can refer to [this script](https://github.com/patil-suraj/muse-experiments/blob/f71e7e79af24509ddb4e1b295a1d0ef8d8758dc9/ctrlnet/train_controlnet_webdataset.py) for full discolsure.
 
 #### 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.
+The model was trained on 3M images from LAION aesthetic 6 plus subset, with batch size of 256 for 50k steps with constant learning rate of 3e-5.
 
 #### 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
+One 8xA100 machine
 
 #### Mixed precision
-fp16
+FP16
 
 #### Additional notes