torch>=1.13
pyyaml>=5.4
tqdm
numpy
matplotlib
tensorboard
scikit-learn
pandas
accelerate
