Thank you for your kind explanation.
I will also refer to the explanation related to zip.
I think integrating the dataloader is an efficient way. I tried, but the data loaders I made have different image sizes and batch sizes. So the integration failed.
In addition, is there a way to learn the input size and batch size differently, or is there an example that I can refer to?
For example,
multi_scale = [(1025, 512), (512, 256), (256, 128)]
multi_batch = [3, 4, 12]
After setting as follows, I would like to configure the batch size according to the input size.
I couldn’t find a way to integrate it so I configured the dataloader differently.
Is there a good way??