From 0b9987f5e9b48c403860cb3b6f1725a495649e70 Mon Sep 17 00:00:00 2001 From: Anna Alberska Date: Thu, 25 Jun 2020 11:43:47 +0200 Subject: [PATCH] [GNA] fix custom scale factors when importing a model (#1096) --- inference-engine/samples/speech_sample/main.cpp | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/inference-engine/samples/speech_sample/main.cpp b/inference-engine/samples/speech_sample/main.cpp index 58b4b98874b..36b405490b4 100644 --- a/inference-engine/samples/speech_sample/main.cpp +++ b/inference-engine/samples/speech_sample/main.cpp @@ -603,12 +603,13 @@ int main(int argc, char *argv[]) { if (FLAGS_q.compare("user") == 0) { auto scaleFactorInput = ParseScaleFactors(FLAGS_sf); - if (scaleFactorInput.size() != network.getInputsInfo().size()) { + if (numInputArkFiles != scaleFactorInput.size()) { std::string errMessage("Incorrect command line for multiple inputs: " - + std::to_string(scaleFactorInput.size()) + " scale factors provided for " - + std::to_string(network.getInputsInfo().size()) + " inputs."); + + std::to_string(scaleFactorInput.size()) + " scale factors provided for " + + std::to_string(numInputArkFiles) + " input files."); throw std::logic_error(errMessage); } + for (size_t i = 0; i < scaleFactorInput.size(); ++i) { slog::info << "For input " << i << " using scale factor of " << scaleFactorInput[i] << slog::endl; std::string scaleFactorConfigKey = GNA_CONFIG_KEY(SCALE_FACTOR) + std::string("_") + std::to_string(i);