fix write error

4 years ago · a414dd8649
parent d2435e35bc
commit a414dd8649
1 changed files with 1 additions and 1 deletions
--- 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,