Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
monticore
EmbeddedMontiArc
generators
CNNArch2Caffe2
Commits
1b9c5e48
Commit
1b9c5e48
authored
Aug 22, 2018
by
Carlos Alfredo Yeverino Rodriguez
Browse files
Set default value for batch_size in CNNCreator.ftl since training parameters can be empty.
parent
477fea7c
Pipeline
#69759
passed with stages
in 2 minutes and 51 seconds
Changes
4
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
src/main/resources/templates/caffe2/CNNCreator.ftl
View file @
1b9c5e48
...
...
@@ -105,8 +105,9 @@ class ${tc.fileNameWithoutEnding}:
sys.exit(1)
def train(self, batch_size,
def train(self, batch_size
=64
,
num_epoch=10,
eval_metric='acc',
optimizer='adam',
optimizer_params=(('learning_rate', 0.001),),
load_checkpoint=True,
...
...
@@ -158,6 +159,7 @@ class ${tc.fileNameWithoutEnding}:
self.module.fit(
train_data=train_iter,
eval_metric=eval_metric,
eval_data=test_iter,
optimizer=optimizer,
optimizer_params=optimizer_params,
...
...
src/test/resources/target_code/CNNCreator_Alexnet.py
View file @
1b9c5e48
...
...
@@ -105,8 +105,9 @@ class CNNCreator_Alexnet:
sys
.
exit
(
1
)
def
train
(
self
,
batch_size
,
def
train
(
self
,
batch_size
=
64
,
num_epoch
=
10
,
eval_metric
=
'acc'
,
optimizer
=
'adam'
,
optimizer_params
=
((
'learning_rate'
,
0.001
),),
load_checkpoint
=
True
,
...
...
@@ -158,6 +159,7 @@ class CNNCreator_Alexnet:
self
.
module
.
fit
(
train_data
=
train_iter
,
eval_metric
=
eval_metric
,
eval_data
=
test_iter
,
optimizer
=
optimizer
,
optimizer_params
=
optimizer_params
,
...
...
src/test/resources/target_code/CNNCreator_CifarClassifierNetwork.py
View file @
1b9c5e48
...
...
@@ -105,8 +105,9 @@ class CNNCreator_CifarClassifierNetwork:
sys
.
exit
(
1
)
def
train
(
self
,
batch_size
,
def
train
(
self
,
batch_size
=
64
,
num_epoch
=
10
,
eval_metric
=
'acc'
,
optimizer
=
'adam'
,
optimizer_params
=
((
'learning_rate'
,
0.001
),),
load_checkpoint
=
True
,
...
...
@@ -158,6 +159,7 @@ class CNNCreator_CifarClassifierNetwork:
self
.
module
.
fit
(
train_data
=
train_iter
,
eval_metric
=
eval_metric
,
eval_data
=
test_iter
,
optimizer
=
optimizer
,
optimizer_params
=
optimizer_params
,
...
...
src/test/resources/target_code/CNNCreator_VGG16.py
View file @
1b9c5e48
...
...
@@ -105,8 +105,9 @@ class CNNCreator_VGG16:
sys
.
exit
(
1
)
def
train
(
self
,
batch_size
,
def
train
(
self
,
batch_size
=
64
,
num_epoch
=
10
,
eval_metric
=
'acc'
,
optimizer
=
'adam'
,
optimizer_params
=
((
'learning_rate'
,
0.001
),),
load_checkpoint
=
True
,
...
...
@@ -158,6 +159,7 @@ class CNNCreator_VGG16:
self
.
module
.
fit
(
train_data
=
train_iter
,
eval_metric
=
eval_metric
,
eval_data
=
test_iter
,
optimizer
=
optimizer
,
optimizer_params
=
optimizer_params
,
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment