bagua>=0.6
torch
tqdm
transformers<=4.11.3
tensorboardX
