click
numpy
sklearn
tqdm
torch
torchvision

[dev]
pytest>=3.7
