requirement.txt

captum==0.6.0
castle==6.1.0
scikit-image==0.21.0
seaborn==0.12.2
torchvision==0.14.1
pytorch-lightning==2.0.7
pytorch-metric-learning==2.3.0
torchaudio==0.13.1
torch==1.13.1
torchvision==0.14.1
tqdm==4.65.0
matplotlib==3.7.1
numba==0.57.1
numpy==1.26.0
scikit-learn==1.3.0
time_interpret==0.3.0
pandas==2.0.3
typing_extensions==4.7.1