torch>=1.8
pytorch-lightning>=1.8
torchvision
tensorguard==1.0.0
path>=16.0

[dev]
pytest
