torchvision==0.6.0a0+82fd1c8
numpy==1.18.2
torch==1.5.0
dataclasses==0.7
tb-nightly