diff --git a/requirements/requirements.txt b/requirements/requirements.txt index ac7d1a3dd472..3f34a97142a8 100644 --- a/requirements/requirements.txt +++ b/requirements/requirements.txt @@ -1,5 +1,6 @@ numpy>=1.18.2 onnx>=1.7.0 +torchmetrics<=0.7.3 pytorch-lightning>=1.3.0 python-dateutil torch