torch
torchvision
numpy
thop
tqdm
