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

* Also fix #243

Make Paddle --save_dir support a directory name (#277)
* Also fix #243
ca5a5ec4 · luotao1 · Yu Yang · 0e1a22d0 · ca5a5ec4
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/trainer/ParamUtil.cpp paddle/trainer/ParamUtil.cpp +3 -0

未找到文件。
--- 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);