tensorboard
torchinfo==1.5.2
torchvision
tqdm