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