get rid of bottleneck on training (tokenization is a lot faster with document iterator