numpy
argparse
tqdm
pandas
torch
torchvision