Commit 6ccdf121 authored by Sebastian N.'s avatar Sebastian N.
Browse files

Fixed another bug in CNNPredictor

parent 976665f7
Pipeline #205598 failed with stages
in 18 seconds
...@@ -22,7 +22,7 @@ public: ...@@ -22,7 +22,7 @@ public:
<#list tc.getStreamInputNames(networkInstruction.body, true) as variable>"data${variable?index}"<#sep>, </#list> <#list tc.getStreamInputNames(networkInstruction.body, true) as variable>"data${variable?index}"<#sep>, </#list>
</#if> </#if>
}; };
const std::vector<std::vector<mx_uint>> input_shapes = {<#list tc.getStreamInputDimensions(networkInstruction.body) as dimensions>{1, ${tc.join(dimensions, ", ")}}<#sep>, </#list>}; const std::vector<std::vector<mx_uint>> input_shapes = {<#list tc.getStreamInputDimensions(networkInstruction.body) as dimensions>{1, ${tc.join(tc.cutDimensions(dimensions), ", ")}}<#sep>, </#list>};
const bool use_gpu = false; const bool use_gpu = false;
PredictorHandle handle; PredictorHandle handle;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment