Fixed test error

b70389d7 · Thomas Michael Timmermanns · d760eed4 · b70389d7
Commit b70389d7 authored Jun 18, 2018 by Thomas Michael Timmermanns
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/test/resources/target_code/CNNPredictor_cifar10_cifar10Classifier_net.h .../target_code/CNNPredictor_cifar10_cifar10Classifier_net.h +3 -3

No files found.
--- a/src/test/resources/target_code/CNNPredictor_cifar10_cifar10Classifier_net.h
+++ b/src/test/resources/target_code/CNNPredictor_cifar10_cifar10Classifier_net.h
@@ -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());