Skip to content
GitLab
Projects
Groups
Snippets
Help
Loading...
Help
What's new
7
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Open sidebar
monticore
EmbeddedMontiArc
generators
CNNArch2MXNet
Commits
f3bc69ee
Commit
f3bc69ee
authored
Aug 05, 2018
by
Svetlana Pavlitskaya
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
added eval-metric
parent
f78e68ea
Pipeline
#66619
passed with stages
in 2 minutes and 37 seconds
Changes
5
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
14 additions
and
0 deletions
+14
-0
.idea/vcs.xml
.idea/vcs.xml
+6
-0
src/main/resources/templates/CNNCreator.ftl
src/main/resources/templates/CNNCreator.ftl
+2
-0
src/test/resources/target_code/CNNCreator_Alexnet.py
src/test/resources/target_code/CNNCreator_Alexnet.py
+2
-0
src/test/resources/target_code/CNNCreator_CifarClassifierNetwork.py
...esources/target_code/CNNCreator_CifarClassifierNetwork.py
+2
-0
src/test/resources/target_code/CNNCreator_VGG16.py
src/test/resources/target_code/CNNCreator_VGG16.py
+2
-0
No files found.
.idea/vcs.xml
0 → 100644
View file @
f3bc69ee
<?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/CNNCreator.ftl
View file @
f3bc69ee
...
...
@@ -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 @
f3bc69ee
...
...
@@ -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 @
f3bc69ee
...
...
@@ -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 @
f3bc69ee
...
...
@@ -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
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a 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