pytest-cov
torch
torchvision
