forked from bstriner/pytorch-igniter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
28 lines (28 loc) · 909 Bytes
/
setup.py
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
from setuptools import find_packages, setup
import os
with open(os.path.abspath(os.path.join(__file__, '../README.rst')), encoding='utf-8') as f:
long_description = f.read()
setup(name='pytorch-igniter',
version='0.0.47',
author='Ben Striner',
author_email="bstriner@gmail.com",
url='https://github.com/bstriner/pytorch-igniter',
description="Simplify running pytorch training with fully-configured pytorch-ignite",
install_requires=[
'pytorch-ignite',
'pyyaml',
'tqdm',
'mlflow',
'aws-sagemaker-remote'
],
package_data={'': [
'**/Dockerfile',
'**/*.sh',
'**/*.txt',
'**/*.yaml',
'**/*.bat'
]},
include_package_data=True,
packages=find_packages(),
long_description=long_description,
long_description_content_type='text/x-rst')