setup.py

from setuptools import setup

__VERSION__ = '0.0.5'

setup(name='adabound',
      version=__VERSION__,
      description='AdaBound optimization algorithm, build on PyTorch.',
      long_description=open("README.md").read(),
      long_description_content_type="text/markdown",
      keywords=['machine learning', 'deep learning'],
      classifiers=[
          'Intended Audience :: Science/Research',
          'Development Status :: 3 - Alpha',
          'License :: OSI Approved :: Apache Software License',
          'Programming Language :: Python :: 3.6',
          'Programming Language :: Python :: 3.7',
          'Topic :: Scientific/Engineering :: Artificial Intelligence',
      ],
      url='https://github.com/Luolc/AdaBound',
      author='Liangchen Luo',
      author_email='luolc.witty@gmail.com',
      license='Apache',
      packages=['adabound'],
      install_requires=[
          'torch>=0.4.0',
      ],
      zip_safe=False,
      python_requires='>=3.6.0')