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
CNNArch2Gluon
Commits
4d88a180
Commit
4d88a180
authored
Aug 20, 2018
by
Carlos Alfredo Yeverino Rodriguez
Browse files
Merge branch 'master' into added-trainer
parents
ce8f15af
29d9750a
Changes
5
Hide whitespace changes
Inline
Side-by-side
.idea/vcs.xml
0 → 100644
View file @
4d88a180
<?xml version="1.0" encoding="UTF-8"?>
<project
version=
"4"
>
<component
name=
"VcsDirectoryMappings"
>
<mapping
directory=
""
vcs=
"Git"
/>
</component>
</project>
\ No newline at end of file
src/main/resources/templates/mxnet/CNNCreator.ftl
View file @
4d88a180
...
...
@@ -107,6 +107,7 @@ class ${tc.fileNameWithoutEnding}:
def train(self, batch_size,
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 @
4d88a180
...
...
@@ -107,6 +107,7 @@ class CNNCreator_Alexnet:
def
train
(
self
,
batch_size
,
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 @
4d88a180
...
...
@@ -107,6 +107,7 @@ class CNNCreator_CifarClassifierNetwork:
def
train
(
self
,
batch_size
,
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 @
4d88a180
...
...
@@ -107,6 +107,7 @@ class CNNCreator_VGG16:
def
train
(
self
,
batch_size
,
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