File size: 586 Bytes
bf80e9a a70873c |
1 2 3 4 5 6 |
Repo for controlnet model trained on 3.5M samples from coyo-700M dataset and 2.6M samples from bridge dataset For each training step, the batch is composed of 28 images from coyo and 4 from Bridge. We created a custom DataLoader that would load the batch with this 28:4 ratio each step from randomly selected images in both dataset. Therefore, we didn't trained for a certain number of epoch but for a certain number of step. If we consider that an epoch correspond to the process of 2.6M images, then we processed 2.275M images from coyo and 325k images from bridge per epoch. |