1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
from setuptools import setup, find_packages
from pathlib import Path
# see https://packaging.python.org/guides/single-sourcing-package-version/
version_dict = {}
with open(Path(__file__).parents[0] / "torch_ema/_version.py") as fp:
exec(fp.read(), version_dict)
version = version_dict["__version__"]
del version_dict
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(),
)
|
