Skip to content

Commit 6925d56

Browse files
committed
update default template
1 parent ea10c5d commit 6925d56

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

modelmaker/resources/templates/default/package/scripts/train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import numpy as np
22
from tensorflow import keras
3-
from default.models import SimpleClassification
4-
from default.utils import generate_square, generate_circle
3+
from {{ package_name }}.models import SimpleClassification
4+
from {{ package_name }}.utils import generate_square, generate_circle
55

66
######################################################################
77
# train SimpleClassification

modelmaker/resources/templates/default/setup.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@
66
version=__version__,
77
author='',
88
install_requires=[
9+
'numpy',
10+
'opencv-python'
911
],
1012
url='',
1113
long_description=open('README.md').read(),

0 commit comments

Comments
 (0)