[BUGFIX] Set minimum required version of pytorch_lightning

The `check_on_train_epoch_end=True` argument to the EarlyStopping callback from pytorch_lightning does not seem to be available in pl version 1.2.8.
2021-06-10 01:14:01 +02:00 · 2021-06-10 01:14:01 +02:00 · 4ca846997a
commit 4ca846997a
parent 57f8bec270
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -21,7 +21,7 @@ with open("README.md", "r") as fh:

 INSTALL_REQUIRES = [
    "prototorch>=0.5.0",
-    "pytorch_lightning",
+    "pytorch_lightning>=1.3.5",
    "torchmetrics",
 ]
 CLI = [