torch
torchvision
numpy
tqdm
pyyaml
