Thanks for writing this interface for RepNet. It's quite neat. We fixed a batch sampling issue in this [commit](https://github.com/google-research/google-research/commit/7a1f1411dd51c480495a6849b995aaa52ace8f38). Please modify the [sampling](https://github.com/JeiKeiLim/repnet-cli/blob/8ae1b117abc07422c92ba13d5b9cafe0eed63bae/repnet/repnet.py#L206) in your client accordingly.
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.