diff --git a/tools/train.py b/tools/train.py index 32abe0fc76a585a00daba6709b5e819952fcd466..c16223796f8785d1928ba1df57602967e0518e31 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,