diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp index e2572872814364cb643ace0a882f8ec984eb4be9..0dc197c3e2a0b7298856129c2316f8f3c6c65561 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; }