diff --git a/requirements.txt b/requirements.txt index 392de7d885..cfa9cfdbfe 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ packaging +numpy torch>=1.7.1 torchmetrics<=0.5.0 pytorch-lightning>=1.4.0