首页 PaddleOCR 帖子详情
训练过程中评估的速度很慢,有什么办法加快吗?
收藏
快速回复
PaddleOCR 问答检测 1168 1
训练过程中评估的速度很慢,有什么办法加快吗?
收藏
快速回复
PaddleOCR 问答检测 1168 1

[2022/03/06 03:27:03] root INFO: epoch: [35/5000], iter: 5990, lr: 0.000100, loss: 1.938344, loss_shrink_maps: 1.091385, loss_threshold_maps: 0.636650, loss_binary_maps: 0.212405, reader_cost: 0.40183 s, batch_cost: 0.65372 s, samples: 80, ips: 12.23763
[2022/03/06 03:27:11] root INFO: epoch: [35/5000], iter: 6000, lr: 0.000100, loss: 2.023747, loss_shrink_maps: 1.144246, loss_threshold_maps: 0.646133, loss_binary_maps: 0.223270, reader_cost: 0.56848 s, batch_cost: 0.81862 s, samples: 80, ips: 9.77254

eval model:: 0%| | 0/5097 [00:00eval model:: 0%| | 1/5097 [00:04<6:19:56, 4.47s/it]
eval model:: 0%| | 2/5097 [00:05<4:56:24, 3.49s/it]

 

用的是v100至尊版

MobileNetV3+DBNet

验证配置

  loader:
    shuffle: False
    drop_last: False
    batch_size_per_card: 1 # must be 1
    num_workers: 8
    use_shared_memory: False

 

0
收藏
回复
全部评论(1)
时间顺序
划飞桨的Daniel
#2 回复于2022-03

评估部分确实耗时较多,如果觉得不需要,也可以注释掉eval()

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户