If I wish to do online hard negative mining (for triplet based algorithms), then I wish to send the recent state dict of my network to the data loader to evaluate the criterion and select the negative. Currently, you need to have a lot of workaround for this to get it working. I think the major limitation is the usage of the getitem directive. Can we have a more flexible approach which allows to send arbitrary number of parameters? Or is there a solution in the current situation which I am not aware of?

Thanks.

cc @ssnl