torch~=1.11.0
torchvision~=0.12.0
bitorchinfo
matplotlib
numpy
sklearn
pytorch_lightning
torchmetrics
