From 79b3be6a42aa06f6265ce7cab97a781433163ee9 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 6 Apr 2020 15:40:03 +0200
Subject: [PATCH] Fixed detail in print
---
trainer/src/MacaonTrain.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/trainer/src/MacaonTrain.cpp b/trainer/src/MacaonTrain.cpp
index 6fdcf03..2f98ff3 100644
--- a/trainer/src/MacaonTrain.cpp
+++ b/trainer/src/MacaonTrain.cpp
@@ -188,7 +188,7 @@ int MacaonTrain::main()
}
machine.saveLast();
trainer.saveOptimizer(optimizerCheckpoint);
- if (!debug)
+ if (printAdvancement)
fmt::print(stderr, "\r{:80}\r", "");
std::string iterStr = fmt::format("[{}] Epoch {:^5} loss = {:6.1f} dev = {} {:5}", util::getTime(), fmt::format("{}/{}", currentEpoch+1, nbEpoch), loss, devScoresStr, saved ? "SAVED" : "");
fmt::print(stderr, "{}\n", iterStr);
--
GitLab