change requirement

3 years ago · 1a80ee315e
1 changed files with 2 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,5 +1,5 @@
-torch==1.9.0
-torchvision==0.10.0
+torch
+torchvision
 pillow==8.3.1
 numpy==1.19.5
 timm==0.4.12