Hi,
I’m currently having a use case of creating custom data loader that can: (i) change batch_size value dynamically during training and (ii) process the data sample with different operation for each different batch_size.
For (i), I successfully accomplish it by creating my custom batch_sampler that returns (a) batch_sample_indices with different number of indices (its len = batch_size). However, I still cannot solve (ii). In order to achieve (ii), I need to know (a), or at least the len of (a) in the __getitem__() method inside the data loader class. Any body can give a direction for this? Or any other approaches to achieve (i) & (ii) is also welcomed.
Thanks in advance!