aboutsummaryrefslogtreecommitdiff
path: root/setup.py
diff options
context:
space:
mode:
Diffstat (limited to 'setup.py')
-rw-r--r--setup.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..8d6fa9e
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,24 @@
+from setuptools import setup, find_packages
+
+__version__ = '0.1'
+url = 'https://github.com/fadel/pytorch_ema'
+download_url = '{}/archive/{}.tar.gz'.format(url, __version__)
+
+install_requires = []
+setup_requires = []
+tests_require = []
+
+setup(
+ name='torch_ema',
+ version=__version__,
+ description='PyTorch library for computing moving averages of model parameters.',
+ author='Samuel G. Fadel',
+ author_email='samuelfadel@gmail.com',
+ url=url,
+ download_url=download_url,
+ keywords=['pytorch', 'parameters', 'deep-learning'],
+ install_requires=install_requires,
+ setup_requires=setup_requires,
+ tests_require=tests_require,
+ packages=find_packages(),
+)