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
5dabf385
Commit
5dabf385
authored
Jun 17, 2018
by
Svetlana
Browse files
Uncommented line in the template file to define correct dataset name.
parent
8ac5c90c
Changes
4
Hide whitespace changes
Inline
Side-by-side
src/main/resources/templates/CNNPredictor.ftl
View file @
5dabf385
...
...
@@ -13,8 +13,8 @@ class ${tc.fileNameWithoutEnding}{
public
:
const
std
::
string
json_file
=
"model/${tc.fullArchitectureName}/${tc.architectureName}_newest-symbol.json"
;
const
std
::
string
param_file
=
"model/${tc.fullArchitectureName}/${tc.architectureName}_newest-0000.params"
;
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
}
;
//
const
std
::
vector
<
std
::
string
>
input_keys
=
{
$
{
tc
.join
(
tc
.architectureInputs
,
","
,
"
\"
"
,
"
\"
"
)}}
;
//
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
}
;
const
std
::
vector
<
std
::
string
>
input_keys
=
{
$
{
tc
.join
(
tc
.architectureInputs
,
","
,
"
\"
"
,
"
\"
"
)}}
;
const
std
::
vector
<
std
::
vector
<
mx_uint
>>
input_shapes
=
{
<#
list
tc
.architecture.inputs
as
input
>
{
1
,
$
{
tc
.join
(
input
.definition.type.dimensions
,
","
)}}
<#
if
input
?
has_next
>
,
</#
if
></#
list
>
}
;
const
bool
use_gpu
=
false
;
...
...
src/test/resources/target_code/CNNPredictor_Alexnet.h
View file @
5dabf385
...
...
@@ -13,8 +13,8 @@ class CNNPredictor_Alexnet{
public:
const
std
::
string
json_file
=
"model/Alexnet/Alexnet_newest-symbol.json"
;
const
std
::
string
param_file
=
"model/Alexnet/Alexnet_newest-0000.params"
;
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
};
//const std::vector<std::string> input_keys = {"data"};
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
};
const
std
::
vector
<
std
::
vector
<
mx_uint
>>
input_shapes
=
{{
1
,
3
,
224
,
224
}};
const
bool
use_gpu
=
false
;
...
...
src/test/resources/target_code/CNNPredictor_CifarClassifierNetwork.h
View file @
5dabf385
...
...
@@ -13,8 +13,8 @@ class CNNPredictor_CifarClassifierNetwork{
public:
const
std
::
string
json_file
=
"model/CifarClassifierNetwork/CifarClassifierNetwork_newest-symbol.json"
;
const
std
::
string
param_file
=
"model/CifarClassifierNetwork/CifarClassifierNetwork_newest-0000.params"
;
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
};
//const std::vector<std::string> input_keys = {"data"};
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
};
const
std
::
vector
<
std
::
vector
<
mx_uint
>>
input_shapes
=
{{
1
,
3
,
32
,
32
}};
const
bool
use_gpu
=
false
;
...
...
src/test/resources/target_code/CNNPredictor_VGG16.h
View file @
5dabf385
...
...
@@ -13,8 +13,8 @@ class CNNPredictor_VGG16{
public:
const
std
::
string
json_file
=
"model/VGG16/VGG16_newest-symbol.json"
;
const
std
::
string
param_file
=
"model/VGG16/VGG16_newest-0000.params"
;
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
};
//const std::vector<std::string> input_keys = {"data"};
const
std
::
vector
<
std
::
string
>
input_keys
=
{
"data"
};
const
std
::
vector
<
std
::
vector
<
mx_uint
>>
input_shapes
=
{{
1
,
3
,
224
,
224
}};
const
bool
use_gpu
=
false
;
...
...
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