added dependencies to pip package
@@ -35,4 +35,14 @@ setuptools.setup(
"Operating System :: OS Independent",
],
python_requires='>=3.6',
+ install_requires=[
+ 'psutil',
+ 'torch',
+ 'torchvision',
+ 'tensorboard',
+ 'scikit-learn',
+ 'matplotlib',
+ 'numpy',
+ 'pytorchcv'
+ ]
)
