Make Paddle --save_dir support a directory name (#277)

* Also fix #243
PaddlePaddle · Oct 28, 2016 · ca5a5ec · ca5a5ec
1 parent 0e1a22d
commit ca5a5ec
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/paddle/trainer/ParamUtil.cpp b/paddle/trainer/ParamUtil.cpp
@@ -89,6 +89,9 @@ void ParameterUtil::saveParameters(int passId, int passInnerId) {
   }
 
   std::string basePath = config_->getSaveDir();
+  if (basePath.find('/') == std::string::npos) {
+    basePath = "./" + basePath;
+  }
   mkDirRecursively(basePath.c_str());
 
   std::string saveDir = path::join(basePath, buf);