language model works with Sampler (not ShuffleSampler)