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
EMADL2CPP
Commits
b70389d7
Commit
b70389d7
authored
Jun 18, 2018
by
Thomas Michael Timmermanns
Browse files
Fixed test error
parent
d760eed4
Pipeline
#56341
passed with stage
in 1 minute and 4 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
src/test/resources/target_code/CNNPredictor_cifar10_cifar10Classifier_net.h
View file @
b70389d7
...
...
@@ -13,8 +13,8 @@ class CNNPredictor_cifar10_cifar10Classifier_net{
public:
const
std
::
string
json_file
=
"model/cifar10_cifar10Classifier_net/net_newest-symbol.json"
;
const
std
::
string
param_file
=
"model/cifar10_cifar10Classifier_net/net_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
;
...
...
@@ -28,8 +28,8 @@ public:
if
(
handle
)
MXPredFree
(
handle
);
}
void
predict
(
const
vector
<
float
>
&
data
,
vector
<
float
>
&
softmax
){
void
predict
(
const
std
::
vector
<
float
>
&
data
,
std
::
vector
<
float
>
&
softmax
){
MXPredSetInput
(
handle
,
"data"
,
data
.
data
(),
data
.
size
());
//MXPredSetInput(handle, "data", data.data(), data.size());
...
...
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