fix write error

This commit is contained in:
WenmuZhou
2020-11-06 19:15:21 +08:00
parent d2435e35bc
commit a414dd8649
+1 -1
View File
@@ -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,