Fixed test error

parent d760eed4
Pipeline #56341 passed with stage
in 1 minute and 4 seconds
......@@ -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());
......
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