Files
ColossalAI/examples/community/roberta/pretraining/bert_dataset_provider.py