Merged PR 13123: Print logs for the model conversion type

Print logs for the model conversion type
shashwatup9k · May 28, 2020 · 1e0e771 · 1e0e771
1 parent 51b13bb
commit 1e0e771
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/command/marian_conv.cpp b/src/command/marian_conv.cpp
@@ -49,7 +49,7 @@ int main(int argc, char** argv) {
     ABORT("Unknown gemm-type: {}", saveGemmTypeStr);
   }
 
-  LOG(info, "Outputting {}", modelTo);
+  LOG(info, "Outputting {}, precision: {}", modelTo, saveGemmType);
 
   YAML::Node config;
   std::stringstream configStr;

diff --git a/src/microsoft/quicksand.cpp b/src/microsoft/quicksand.cpp
@@ -246,7 +246,7 @@ DecoderCpuAvxVersion parseCpuAvxVersion(std::string name) {
 
 // @TODO: clean-up this code and unify with marian-conv. The targetPrec parameter is not clear enought etc.
 bool convertModel(std::string inputFile, std::string outputFile, int32_t targetPrec) {
-  std::cout << "Converting from: " << inputFile << ", to: " << outputFile << std::endl;
+  std::cerr << "Converting from: " << inputFile << ", to: " << outputFile << ", precision: " << targetPrec << std::endl;
 
   YAML::Node config;
   std::stringstream configStr;
@@ -268,7 +268,7 @@ bool convertModel(std::string inputFile, std::string outputFile, int32_t targetP
   // added a flag if the weights needs to be packed or not
   graph->packAndSave(outputFile, configStr.str(), saveGemmType);
 
-  std::cout << "Conversion Finished." << std::endl;
+  std::cerr << "Conversion Finished." << std::endl;
 
   return true;
 }