PaddleHub icon indicating copy to clipboard operation
PaddleHub copied to clipboard

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

Open zhxnlp opened this issue 3 years ago • 1 comments

欢迎您反馈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左右)在这里插入图片描述 3. 我后来仔细看了那个桃子分拣教程,教程里面,‘Evaluation on validation dataset’也是重复打印多次。希望可以改进

zhxnlp avatar Sep 14 '22 13:09 zhxnlp

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

rainyfly avatar Sep 16 '22 03:09 rainyfly