SSLab/setup.py at master · DSciLab/SSLab · 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
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
from distutils.core import setup
import datetime


def gen_code():
    d = datetime.datetime.now()
    date_str = d.strftime('%Y%m%d%H%M%S')

    return f'dev{date_str}'


__version__ = f'0.0.1.{gen_code()}'


setup(name='ssl_lab',
      version=__version__,
      description='Self-Supervised Lab',
      author='tor4z',
      author_email='vwenjie@hotmail.com',
      install_requires=[
            'torch',
            'einops',
            'numpy',
            'timm',
            'ml-collections',
            'jupyterlab',
            'matplotlib',
      ],
      packages=['ssl_lab',
                'ssl_lab.dataset',
                'ssl_lab.network',
                'ssl_lab.network.resnet',
                'ssl_lab.network.vit',
                'ssl_lab.network.byol',
                'ssl_lab.network.dino',
                'ssl_lab.network.mocov3',
                'ssl_lab.network.mocov2',
                'ssl_lab.network.mocov1',
                'ssl_lab.network.simclrv1',
                # trainer
                'ssl_lab.trainer',
                'ssl_lab.trainer.resnet',
                'ssl_lab.trainer.vit',
                'ssl_lab.trainer.byol',
                'ssl_lab.trainer.dino',
                'ssl_lab.trainer.mocov3',
                'ssl_lab.trainer.mocov2',
                'ssl_lab.trainer.mocov1',
                'ssl_lab.trainer.simclrv1',
                'ssl_lab.utils',
                'ssl_lab.utils.data'
      ]
)