dat/setup.py at master · Capsar/dat · GitHub

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

setup(
    name='dat-jointspar-trainer',
    version='0.3',
    packages=find_packages(),
    include_package_data=True,
    install_requires=[
        'torch',
        'torchvision',
        'numpy',
        'tqdm',
        'requests',
        'wandb',
        'tensorboard',
        'python-dotenv'
    ],
    # entry_points={
    #         'console_scripts': [
    #             'dat_package=trainer.task:main',
    #         ],
    # },
    description='Training code application for DAT (Distributed Adversarial Training). Based on paper of Distributed Adversarial Training to Robustify Deep Neural Networks at Scale.'
)