Custom Sampler in Pytorch

[kris-singh]

Resolved here. https://github.com/pytorch/pytorch/issues/7359