From a618534ebcc9dd4af735afac9459b075a63367f7 Mon Sep 17 00:00:00 2001 From: flytocc Date: Mon, 9 May 2022 11:51:26 +0800 Subject: [PATCH] fix log_info --- ppcls/engine/train/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/engine/train/utils.py b/ppcls/engine/train/utils.py index 7f64104d..a54d7a46 100644 --- a/ppcls/engine/train/utils.py +++ b/ppcls/engine/train/utils.py @@ -54,12 +54,12 @@ def log_info(trainer, batch_size, epoch_id, iter_id): ips_msg = "ips: {:.5f} samples/s".format( batch_size / trainer.time_info["batch_cost"].avg) eta_sec = ((trainer.config["Global"]["epochs"] - epoch_id + 1 - ) * len(trainer.train_dataloader) - iter_id + ) * trainer.max_iter - iter_id ) * trainer.time_info["batch_cost"].avg eta_msg = "eta: {:s}".format(str(datetime.timedelta(seconds=int(eta_sec)))) logger.info("[Train][Epoch {}/{}][Iter: {}/{}]{}, {}, {}, {}, {}".format( epoch_id, trainer.config["Global"]["epochs"], iter_id, - len(trainer.train_dataloader), lr_msg, metric_msg, time_msg, ips_msg, + trainer.max_iter, lr_msg, metric_msg, time_msg, ips_msg, eta_msg)) for i, lr in enumerate(trainer.lr_sch): -- GitLab