Commit 383759ae by xiaotong

remove model dumping

parent af3acf68
...@@ -83,7 +83,7 @@ int TransformerMain(int argc, const char ** argv) ...@@ -83,7 +83,7 @@ int TransformerMain(int argc, const char ** argv)
model.InitModel(argc, args); model.InitModel(argc, args);
isTrain = trainer.Train(trainFN, testFN, modelFN, &model); isTrain = trainer.Train(trainFN, testFN, modelFN, &model);
model.Dump(fn1); //model.Dump(fn1);
} }
else { else {
T2TModel model; T2TModel model;
...@@ -91,7 +91,7 @@ int TransformerMain(int argc, const char ** argv) ...@@ -91,7 +91,7 @@ int TransformerMain(int argc, const char ** argv)
model.Read(fn); model.Read(fn);
isTrain = trainer.Train(trainFN, testFN, modelFN, &model); isTrain = trainer.Train(trainFN, testFN, modelFN, &model);
model.Dump(fn1); //model.Dump(fn1);
} }
if(trainer.useEpochCheckpoint && strcmp(testFN, "")) { if(trainer.useEpochCheckpoint && strcmp(testFN, "")) {
...@@ -100,9 +100,9 @@ int TransformerMain(int argc, const char ** argv) ...@@ -100,9 +100,9 @@ int TransformerMain(int argc, const char ** argv)
T2TModel model; T2TModel model;
model.InitModel(argc, args); model.InitModel(argc, args);
model.Read(fn1); //model.Read(fn1);
tester.Test(testFN, fn2, &model); //tester.Test(testFN, fn2, &model);
} }
if(!isTrain) if(!isTrain)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论