pytest
pytest-cov
torch
torchvision
tqdm
PyYAML
scikit-learn
