torch
torchvision
tqdm
fvcore
matplotlib
einops
