diff --git a/requirements/torch.txt b/requirements/torch.txt index 617ef86948..b38e319e03 100644 --- a/requirements/torch.txt +++ b/requirements/torch.txt @@ -1,3 +1,3 @@ -pytorch-lightning>=1.5.0 +pytorch-lightning>=1.5.0,<=2.1.2 tensorboardX>=2.1 torch>=1.8.0