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(),
+)