diff --git a/requirements.txt b/requirements.txt index 83cc79a..6c818b7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,7 +10,7 @@ imageio[ffmpeg] imageio[pyav] # correction CNN training requirements -torchmetrics==0.8.2 # DO NOT change version here, otherwise it will break the training (complaints about Accuracy()) +torchmetrics==1.4.0.post0 # DO NOT change version here, otherwise it will break the training (complaints about Accuracy()) # labelme2coco requirements below sahi>=0.8.19