torch
torchvision
tqdm
PyYAML
