PaddlePaddle / PaddleHub

Awesome pre-trained models toolkit based on PaddlePaddle. (400+ models including Image, Text, Audio, Video and Cross-Modal with Easy Inference & Serving)【安全加固,暂停交互,请耐心等待】
https://www.paddlepaddle.org.cn/hub
Apache License 2.0
12.75k stars 2.07k forks source link

trainer训练时验证结果重复打印太多次,导致打印日志很难看清 #2024

Open zhxnlp opened 2 years ago

zhxnlp commented 2 years ago

欢迎您反馈PaddleHub使用问题,非常感谢您对PaddleHub的贡献! 在留下您的问题时,辛苦您同步提供如下信息:

  1. paddlehub版本2.3.0,代码在项目中公开,《华为车道检测》。主要代码:

import paddlehub as hub model = hub.Module(name="resnet50_vd_imagenet_ssld", label_list=['0','1'])

import paddle from paddlehub.finetune.trainer import Trainer

optimizer = paddle.optimizer.Adam(learning_rate=0.001, parameters=model.parameters()) trainer = Trainer(model, optimizer, checkpoint_dir='resnetv2_50') trainer.train(train_ds, epochs=1, batch_size=32, eval_dataset=valid_ds, save_interval=1)

运行后训练时日志打印正常,验证集明显不对,每次验证,‘Evaluation on validation dataset’重复打印520+次,导致日志很难看loss等信息。(验证集dataset长度200左右)在这里插入图片描述

  1. 我后来仔细看了那个桃子分拣教程,教程里面,‘Evaluation on validation dataset’也是重复打印多次。希望可以改进
rainyfly commented 2 years ago

你好,问题的原因已经找到了,当初在执行验证集eval的时候,做了个打印Evaluation on validation dataset的等待动画,正常应该是会打印在同一行,但是在aistudio上打印不生效,我们看下如何优化。