Merge pull request #3810 from lcy-seso/fix_print_layer

fix the problem that long message will be truncated.
PaddlePaddle · Sep 1, 2017 · fc8a1af · fc8a1af
2 parents 2d31ab5 + ba1aa60
commit fc8a1af
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/paddle/gserver/layers/PrintLayer.cpp b/paddle/gserver/layers/PrintLayer.cpp
@@ -48,7 +48,16 @@ class PrintLayer : public Layer {
                  << inputLayers_.size() << ") at " << getName();
     }
     s << format.substr(pos);
-    LOG(INFO) << s.str();
+
+    const std::string delimiter("\n");
+    std::string content = s.str();
+    std::string::size_type foundPos = 0;
+    std::string::size_type prevPos = 0;
+    while ((foundPos = content.find(delimiter, prevPos)) != std::string::npos) {
+      LOG(INFO) << content.substr(prevPos, foundPos - prevPos);
+      prevPos = foundPos + delimiter.size();
+    }
+    LOG(INFO) << content.substr(prevPos);
   }
 
   void backward(const UpdateCallback& callback) override {}