johnbroughton2017 opened a new issue #12814: MXPredReshape always fails (return -1) - MXNet 1.3.0 c++ URL: https://github.com/apache/incubator-mxnet/issues/12814 Hi friends, I am having some problem with MXPredReshape(). I tried to reshape the input to a larger size but the function always returns -1. The code is given below. I am not sure what went wrong since there is not error hints. Any suggestion is highly appreciated. Thanks. -- J ` // setup input data mx_uint num_input_nodes = 1; // 1 for feedforward const char* input_key[1] = {"data"}; const char** input_keys = input_key; const mx_uint input_shape_indptr[2] = { 0, 4 }; const mx_uint input_shape_data[4] = { batch_size, input_depth, input_height, input_width }; // create Predictor MXPredCreatePartialOut( (const char*)model_buf.GetBuffer(), (const char*)param_buf.GetBuffer(), static_cast<size_t >(param_buf.GetLength()), device_type, device_id, num_input_nodes, input_keys, input_shape_indptr, input_shape_data, 0, NULL, &pred_hnd); if (!pred_hnd) { LOG(ERROR) << "Fail to create MXNet predictor"; return false; } PredictorHandle reshaped_pred_hnd = NULL; const mx_uint new_input_shape_data[4] = { batch_size, input_depth, 512, 512 }; int ret = MXPredReshape(num_input_nodes, input_keys, input_shape_indptr, new_input_shape_data, pred_hnd, &reshaped_pred_hnd); `
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
