[install] Pin pytorch-optimizer version to support RAdam (#568)

In jettify/pytorch-optimizer#377 the RAdam optimizer was removed from pytorch-optimizer, it's still used in asteroid though.
asteroid-team · Oct 28, 2021 · d21a188 · d21a188
1 parent 9d59ecb
commit d21a188
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements/install.txt b/requirements/install.txt
@@ -6,5 +6,5 @@ pytorch-lightning>=1.0.1
 torchaudio>=0.8.0
 pb_bss_eval>=0.0.2
 torch_stoi>=0.0.1
-torch_optimizer>=0.0.1a12
+torch_optimizer>=0.0.1a12,<0.2.0
 julius
diff --git a/setup.py b/setup.py
@@ -50,7 +50,7 @@ def find_version(*file_paths):
         "torchaudio>=0.5.0",
         "pb_bss_eval>=0.0.2",
         "torch_stoi>=0.1.2",
-        "torch_optimizer>=0.0.1a12",
+        "torch_optimizer>=0.0.1a12,<0.2.0",
         "julius",
     ],
     entry_points={