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

[dev]
pytest
