torch>=1.1.0
torchvision
dataclasses
matplotlib
numpy
pathlib
varname
brokenaxes