crlandsc commited on
Commit
c60d793
1 Parent(s): 7ccb5b3

fixed torchmetrics dependency version issue

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- torch>=2.0
2
- torchaudio>=2.0
3
  pytorch-lightning==1.7.7
4
- torchmetrics
5
  python-dotenv
6
  hydra-core
7
  hydra-colorlog
 
1
+ torch>=2.0.1
2
+ torchaudio>=2.0.2
3
  pytorch-lightning==1.7.7
4
+ torchmetrics==0.11.4
5
  python-dotenv
6
  hydra-core
7
  hydra-colorlog