Improved the performance of BernoulliSampler
Update data loaing benchmark