From 54867a47a7e0e1fb87dc9ff15808ea6eb5551221 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Thu, 5 Mar 2020 21:46:45 +0100
Subject: [PATCH] Do not print advancement when decoding dev
---
trainer/src/Trainer.cpp | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp
index e257287..0dc197c 100644
--- a/trainer/src/Trainer.cpp
+++ b/trainer/src/Trainer.cpp
@@ -133,7 +133,10 @@ float Trainer::processDataset(DataLoader & loader, bool train, bool printAdvance
auto actualTime = std::chrono::high_resolution_clock::now();
double seconds = std::chrono::duration<double, std::milli>(actualTime-pastTime).count() / 1000.0;
pastTime = actualTime;
- fmt::print(stderr, "\rcurrent epoch : {:6.2f}% loss={:<7.3f} speed={:<5}ex/s", 100.0*(currentBatchNumber*batchSize)/nbExamples, lossSoFar, (int)(nbExamplesProcessed/seconds));
+ if (train)
+ fmt::print(stderr, "\rcurrent epoch : {:6.2f}% loss={:<7.3f} speed={:<6}ex/s", 100.0*(currentBatchNumber*batchSize)/nbExamples, lossSoFar, (int)(nbExamplesProcessed/seconds));
+ else
+ fmt::print(stderr, "\reval on dev : loss={:<7.3f} speed={:<6}ex/s", lossSoFar, (int)(nbExamplesProcessed/seconds));
lossSoFar = 0;
nbExamplesProcessed = 0;
}
--
GitLab