Hi, it’s a relief I am not the only one facing this issue. Thank you so much for posting this! However, I’ve some trouble understanding how you solved this. Can you really have multiple samplers, one per worker? And how do you achieve every worker getting all the indices of one batch and not just a subset?
Maybe you could even share some minimal code? I’ve detailed my current problem and thoughts in this new thread