Pinning PTL to 1.4.2

Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
This commit is contained in:
Shrinath Suresh 2021-08-13 18:30:35 +05:30
parent 3153d8bef3
commit 8e207f0d3b
1 changed files with 1 additions and 1 deletions

View File

@ -24,7 +24,7 @@ from setuptools import setup, find_packages
def make_required_install_packages():
return [
"pytorch-lightning>=1.4.0",
"pytorch-lightning==1.4.2",
"torch>=1.7.1",
"torch-model-archiver",
]