release code
This commit is contained in:
48
Dassl.ProGrad.pytorch/setup.py
Normal file
48
Dassl.ProGrad.pytorch/setup.py
Normal file
@@ -0,0 +1,48 @@
|
||||
import numpy as np
|
||||
import os.path as osp
|
||||
from setuptools import setup, find_packages
|
||||
|
||||
|
||||
def readme():
|
||||
with open('README.md') as f:
|
||||
content = f.read()
|
||||
return content
|
||||
|
||||
|
||||
def find_version():
|
||||
version_file = 'dassl/__init__.py'
|
||||
with open(version_file, 'r') as f:
|
||||
exec(compile(f.read(), version_file, 'exec'))
|
||||
return locals()['__version__']
|
||||
|
||||
|
||||
def numpy_include():
|
||||
try:
|
||||
numpy_include = np.get_include()
|
||||
except AttributeError:
|
||||
numpy_include = np.get_numpy_include()
|
||||
return numpy_include
|
||||
|
||||
|
||||
def get_requirements(filename='requirements.txt'):
|
||||
here = osp.dirname(osp.realpath(__file__))
|
||||
with open(osp.join(here, filename), 'r') as f:
|
||||
requires = [line.replace('\n', '') for line in f.readlines()]
|
||||
return requires
|
||||
|
||||
|
||||
setup(
|
||||
name='dassl',
|
||||
version=find_version(),
|
||||
description='Dassl: Domain adaptation and semi-supervised learning',
|
||||
author='Kaiyang Zhou',
|
||||
license='MIT',
|
||||
long_description=readme(),
|
||||
url='https://github.com/KaiyangZhou/Dassl.pytorch',
|
||||
packages=find_packages(),
|
||||
install_requires=get_requirements(),
|
||||
keywords=[
|
||||
'Domain Adaptation', 'Domain Generalization',
|
||||
'Semi-Supervised Learning', 'Pytorch'
|
||||
]
|
||||
)
|
||||
Reference in New Issue
Block a user