Click>=7.0
torch>=1.0
torchvision
pillow
tensorboard

[:python_version < "3.7"]
dataclasses
