Update README.md

README.md

@@ -60,7 +60,7 @@ license: creativeml-openrail-m
   </p>
   <p style="font-size: 18px; color: #666;">
     <strong>Technical: </strong>
-    <span>The model was trained on 23,088 samples for 90 epochs  (500,000+ steps) using a stepped cosine and stepped constant LR. 60 Epochs of training were done at 768 resolution. 512 and 768 were both using aspect ratio buckets to prevent unnatural cropping.</span>
+    <span>The model was trained on 23,088 samples from Yande.re using a base model NAI, with a resolution of 768 and 512 pixels using aspect ratio buckets to prevent unnatural cropping. The model was trained for 90 epochs (500,000+ steps) using a stepped cosine and stepped constant LR. The LR was manually stepped every 10 epochs and was estimated using proportionality when estimating LR for each additional batch size. The training was done on an RTX 4090 with EveryDream 2 Trainer using DDIM sample scheduler and DDPM noise scheduler, with Adam8bit optimizer, mix precision, and xformers enabled. A conditional dropout of tags was set to 0.125. Verification testing was done on various milestones, and logs were monitored via Tensorboard. </span>
   </p>
   <p style="font-size: 18px; color: #666;">
     <strong>Authors: </strong><span>Juusoz, 金Goldkoron, tsmkirby</span>