pytest
torchvision
tensorboardx
