pytest
torchvision