torchvision==0.13.1
torchaudio==0.12.1
scikit-learn
torchsummary
easydict
tensorboard
timm
nnunet
transformers
thop
wandb
intel_extension_for_pytorch==1.12.100
