ERABB / requirements.txt
909ahmed's picture
init commit
0f13a98
torch==2.1.2
torchvision
torch-lr-finder
grad-cam
pillow
numpy
pytorch_lightning==1.9.5
lightning-bolts