Hello @ptrblck
Your solution works well. I want to know how to set batch size in your this customized Dateset class?
for example: I want set batch-size as 5, so that I can get 5 images during the training process?
Thanks