Skip to content

GitLab

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

Fixed another bug in CNNPredictor

parent 976665f7
Pipeline #205598 failed with stages
      in 18 seconds
      Hide whitespace changes
      Inline Side-by-side
      Showing with 1 addition and 1 deletion
      src/main/resources/templates/gluon/CNNPredictor.ftl
      View file @ 6ccdf121
      ......@@ -22,7 +22,7 @@ public:
      <#list tc.getStreamInputNames(networkInstruction.body, true) as variable>"data${variable?index}"<#sep>, </#list>
      </#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;
      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