Modifying batch size during training

[paganpasta]

Is this something you are looking for ?

Dataloader for variable batch size

Hi I am new to this and for most application I have been using the dataloader in utils.data to load in batches of images. However I am now trying to load images in different batch size. For example my first iteration loads in batch of 10, second loads in batch of 20. Is there a way to do this easily? Thank you.