diff --git a/setup.cfg b/setup.cfg index 28b49cf8..36a23ef8 100644 --- a/setup.cfg +++ b/setup.cfg @@ -52,7 +52,7 @@ install_requires = torchvision>=0.8.1 torch>=1.8 runstats>=1.8 - pytorch-lightning>=1.4 + pytorch-lightning h5py>=2.10 PyYAML>=5.3.1 torchmetrics>=0.5.1