torch
numpy
more_itertools
collectionish>=0.5.0
torchvision
matplotlib
