torch>=2.0.0
torchvision>=0.15.0
perforatedai
wandb
argparse