From 1759bd0b4ff89b6181d84f037ffbeed73236cb77 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Tue, 4 Feb 2020 10:08:46 +0100
Subject: [PATCH] Made print more pretty
---
dev/src/dev.cpp | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/dev/src/dev.cpp b/dev/src/dev.cpp
index 0d9738d..219bacb 100644
--- a/dev/src/dev.cpp
+++ b/dev/src/dev.cpp
@@ -29,10 +29,13 @@ int main(int argc, char * argv[])
Trainer trainer(machine);
trainer.createDataset(config);
- for (int i = 0; i < 5; i++)
+ int nbEpoch = 5;
+
+ for (int i = 0; i < nbEpoch; i++)
{
float loss = trainer.epoch();
- fmt::print("\nEpoch {} loss = {}\n", i+1, loss);
+ fmt::print("\r{:80}", " ");
+ fmt::print("\rEpoch {}/{} loss = {}\n", i+1, nbEpoch, loss);
}
return 0;
--
GitLab