janakg's picture
base
f64a3c9
raw
history blame contribute delete
172 Bytes
# requirements.txt
torch==2.0.1
torchvision==0.15.2
torch-lr-finder==0.2.1
albumentations==1.3.1
torchsummary==1.5.1
grad-cam==1.4.8
pytorch_lightning==2.0.6
gradio==3.39.0