1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
from setuptools import setup, find_packages
__version__ = '0.2'
url = 'https://github.com/fadel/pytorch_ema'
download_url = '{}/archive/{}.tar.gz'.format(url, __version__)
install_requires = ["torch"]
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(),
)
|
