generated from ashleve/lightning-hydra-template
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathsetup.py
21 lines (19 loc) · 825 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/usr/bin/env python
from setuptools import find_packages, setup
setup(
name="pest_rec",
version="0.0.1",
description="Implementation of an advanced Convolutional Neural Network (CNN) for large-scale pest recognition, incorporating augmentation techniques and regularizers for improved accuracy and generalization.",
author="Adhi Setiawan",
author_email="[email protected]",
url="https://github.com/adhiiisetiawan/large-scale-pest-recognition",
install_requires=["lightning", "hydra-core"],
packages=find_packages(),
# use this to customize global commands available in the terminal after installing the package
entry_points={
"console_scripts": [
"train_command = pest_rec.train:main",
"eval_command = pest_rec.eval:main",
]
},
)