lightning>=2.4.0
hydra-core>=1.3
torchvision>=0.19.0
torchmetrics
captum
tensorboard
wandb
pandas
matplotlib
pyrootutils
hydra-colorlog
rich
psutil
typer
gdown
scipy
tinyimagenet
rarfile
