我将参数发送到tensorflow,如下所示:
std::string input_layer = "input_node";
std::string output_layer = "output_node";
std::string dropout_layer = "keep_prob"; //--- extra parameter
Status run_status = session->Run({{input_layer, resized_tensor}},
{output_layer}, {}, &outputs);
if (!run_status.ok()) {
LOG(ERROR) << "Running model failed: " << run_status;
return -1;
}
我试过
tensorflow::Status run_status = session->Run({{input_layer,dropout_layer, image_tensor}},{output_layer},{}, &outputs);
但它不接受额外的参数 如何发送额外的输入参数?