torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 numpy>=1.16.5 scipy>=1.3.0 opencv-python gdown pillow gradio timm==0.4.12 wandb matplotlib
python train.py