tensorboard>=2.20.0
setuptools<80.9
tqdm>=4.66.0

[torch]
torch
torchvision
