diff --git a/requirements.txt b/requirements.txt index 954dd741..a775c8ed 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ numpy>=1.14.2 -torch==0.4.0 +torch>=0.4.0 torchvision>=0.1.8 tensorboardX