when i have developed this concept in my dataset than at the time of batch formation only smaller size data is consider in batch formation.
e.g  data folder i have two subfolders cat1 and cat2. in cat1  70 images and cat2 30 images . Dataloader function make only 30 image considered in batch formation. 40 images of cat2 is not considered in batch formation.? can you have any solution which consider remaning image also