-
Notifications
You must be signed in to change notification settings - Fork 98
/
setup.py
24 lines (22 loc) · 814 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
from setuptools import setup
with open("requirements.txt", "r") as file_:
project_requirements = file_.read().split("\n")
setup(
name="pro-gan-pth",
version="3.4",
packages=["pro_gan_pytorch", "pro_gan_pytorch_scripts"],
url="https://github.com/akanimax/pro_gan_pytorch",
license="MIT",
author="akanimax",
author_email="[email protected]",
setup_requires=['wheel'],
description="ProGAN package implemented as an extension of PyTorch nn.Module",
install_requires=project_requirements,
entry_points={
"console_scripts": [
f"progan_train=pro_gan_pytorch_scripts.train:main",
f"progan_lsid=pro_gan_pytorch_scripts.latent_space_interpolation:main",
f"progan_fid=pro_gan_pytorch_scripts.compute_fid:main",
]
},
)