chendl's picture
add requirements
a1d409e
torch>=1.5.0
torchvision>=0.6.0
datasets>=1.8.0