Thanks for writing this interface for RepNet. It's quite neat.
We fixed a batch sampling issue in this commit. Please modify the sampling in your client accordingly.