diff --git a/requirements.txt b/requirements.txt index c3aa856..19f75ae 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ simclr torchaudio-augmentations -torch==1.9.0 +torch==2.2.0 torchaudio pytorch-lightning soundfile diff --git a/setup.py b/setup.py index d9359c9..7471039 100644 --- a/setup.py +++ b/setup.py @@ -22,7 +22,7 @@ # What packages are required for this module to be executed? REQUIRED = [ - "torch==1.9.0", + "torch==2.2.0", "torchaudio", "simclr", "torchaudio-augmentations",