From a414dd864929a37eed26d153a7351d74e2d678ec Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Fri, 6 Nov 2020 19:15:21 +0800 Subject: [PATCH] fix write error --- tools/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/train.py b/tools/train.py index 32abe0fc..c1622379 100755 --- a/tools/train.py +++ b/tools/train.py @@ -83,7 +83,7 @@ def main(config, device, logger, vdl_writer): # load pretrain model pre_best_model_dict = init_model(config, model, logger, optimizer) - logger.info('train dataloader has {} iter, valid dataloader has {} iter'. + logger.info('train dataloader has {} iters, valid dataloader has {} iters'. format(len(train_dataloader), len(valid_dataloader))) # start train program.train(config, train_dataloader, valid_dataloader, device, model, -- GitLab