tensorboard>=2.20.0
tqdm>=4.66.0

[torch]
torch
torchvision
