diff --git a/requirements.txt b/requirements.txt index f7bc952d9..9c483ac16 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ PyYAML -torch>=1.9.0,<2.0.0 +torch>=1.9.0 numpy>=1.17 scipy>=1.2.1 prefetch_generator>=1.0.1