diff --git a/dev/src/dev.cpp b/dev/src/dev.cpp index 0d9738dbeafcadc3d0cbcf7591335c0fdaa8cbec..219bacb258ec7168bef3549278ea9779346c0475 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;