Given -m, use requested model name for output file

clab · Jul 30, 2015 · 82ad4b4 · 82ad4b4
1 parent 1c557b0
commit 82ad4b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parser/lstm-parse.cc b/parser/lstm-parse.cc
@@ -558,7 +558,7 @@ int main(int argc, char** argv) {
      << '_' << REL_DIM
      << "-pid" << getpid() << ".params";
   int best_correct_heads = 0;
-  const string fname = os.str();
+  const string fname = conf.count("model") ? conf["model"].as<string>() : os.str();
   cerr << "Writing parameters to file: " << fname << endl;
   bool softlinkCreated = false;
   corpus.load_correct_actions(conf["training_data"].as<string>());