torch>=1.7.1
torchmetrics>=0.4.1
pytorch-lightning>=1.4.0
packaging
