tensorboard>=2.21.0
tqdm>=4.66.0

[torch]
torch
torchvision
