torch>=2.0.0
torchvision>=0.15.0
tqdm>=4.65.0
scikit-learn>=1.3.0
matplotlib>=3.7.0
PyYAML>=6.0
wandb>=0.15.0
