mirror of https://github.com/PyRetri/PyRetri.git
upload
parent
ae9a335e2b
commit
47719f1137
2
setup.py
2
setup.py
|
@ -192,7 +192,7 @@ if __name__ == '__main__':
|
|||
],
|
||||
license='Apache License 2.0',
|
||||
install_requires=[
|
||||
'numpy', 'torch>=1.1', 'torchvision>=0.4', 'sklearn', 'yacs', 'tqdm',
|
||||
'numpy', 'torch>=1.2', 'torchvision>=0.4', 'sklearn', 'yacs', 'tqdm',
|
||||
],
|
||||
packages=find_packages(),
|
||||
zip_safe=False)
|
||||
|
|
Loading…
Reference in New Issue