How does one obtain indicies from a dataloader?

See here for a general solution: How to retrieve the sample indices of a mini-batch

2 Likes