pull/1/head
hubenyi 2020-04-21 23:03:25 +08:00
parent ae9a335e2b
commit 47719f1137
1 changed files with 1 additions and 1 deletions

View File

@ -192,7 +192,7 @@ if __name__ == '__main__':
], ],
license='Apache License 2.0', license='Apache License 2.0',
install_requires=[ 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(), packages=find_packages(),
zip_safe=False) zip_safe=False)