YOLOv4-pytorch icon indicating copy to clipboard operation
YOLOv4-pytorch copied to clipboard

训练特别慢

Open 442616159 opened this issue 3 years ago • 2 comments

为什么我训练特别慢,一个epoch要好久,而且学习率为什么显示为0啊

[2021-04-18 21:24:16,560]-[train.py line:231]: === Epoch:[ 0/10],step:[490/562],img_size:[416],total_loss:178.2688|loss_ciou:31.8549|loss_conf:129.4508|loss_cls:16.9630|lr:0.0000 [2021-04-18 21:24:33,861]-[train.py line:231]: === Epoch:[ 0/10],step:[500/562],img_size:[416],total_loss:176.0515|loss_ciou:31.8598|loss_conf:127.4119|loss_cls:16.7799|lr:0.0000 [2021-04-18 21:24:52,436]-[train.py line:231]: === Epoch:[ 0/10],step:[510/562],img_size:[416],total_loss:173.9491|loss_ciou:31.8817|loss_conf:125.4699|loss_cls:16.5975|lr:0.0000 [2021-04-18 21:25:11,434]-[train.py line:231]: === Epoch:[ 0/10],step:[520/562],img_size:[416],total_loss:171.9408|loss_ciou:31.9154|loss_conf:123.6213|loss_cls:16.4042|lr:0.0000 [2021-04-18 21:25:28,948]-[train.py line:231]: === Epoch:[ 0/10],step:[530/562],img_size:[416],total_loss:169.7509|loss_ciou:31.8272|loss_conf:121.7585|loss_cls:16.1653|lr:0.0000 [2021-04-18 21:25:46,376]-[train.py line:231]: === Epoch:[ 0/10],step:[540/562],img_size:[416],total_loss:167.5829|loss_ciou:31.7232|loss_conf:119.9403|loss_cls:15.9195|lr:0.0000 [2021-04-18 21:26:04,282]-[train.py line:231]: === Epoch:[ 0/10],step:[550/562],img_size:[416],total_loss:165.4504|loss_ciou:31.6022|loss_conf:118.1677|loss_cls:15.6805|lr:0.0000 [2021-04-18 21:26:21,893]-[train.py line:231]: === Epoch:[ 0/10],step:[560/562],img_size:[416],total_loss:163.4508|loss_ciou:31.5294|loss_conf:116.4804|loss_cls:15.4410|lr:0.0000

442616159 avatar Apr 18 '21 13:04 442616159

为什么我训练特别慢,一个epoch要好久,而且学习率为什么显示为0啊

[2021-04-18 21:24:16,560]-[train.py line:231]: === Epoch:[ 0/10],step:[490/562],img_size:[416],total_loss:178.2688|loss_ciou:31.8549|loss_conf:129.4508|loss_cls:16.9630|lr:0.0000 [2021-04-18 21:24:33,861]-[train.py line:231]: === Epoch:[ 0/10],step:[500/562],img_size:[416],total_loss:176.0515|loss_ciou:31.8598|loss_conf:127.4119|loss_cls:16.7799|lr:0.0000 [2021-04-18 21:24:52,436]-[train.py line:231]: === Epoch:[ 0/10],step:[510/562],img_size:[416],total_loss:173.9491|loss_ciou:31.8817|loss_conf:125.4699|loss_cls:16.5975|lr:0.0000 [2021-04-18 21:25:11,434]-[train.py line:231]: === Epoch:[ 0/10],step:[520/562],img_size:[416],total_loss:171.9408|loss_ciou:31.9154|loss_conf:123.6213|loss_cls:16.4042|lr:0.0000 [2021-04-18 21:25:28,948]-[train.py line:231]: === Epoch:[ 0/10],step:[530/562],img_size:[416],total_loss:169.7509|loss_ciou:31.8272|loss_conf:121.7585|loss_cls:16.1653|lr:0.0000 [2021-04-18 21:25:46,376]-[train.py line:231]: === Epoch:[ 0/10],step:[540/562],img_size:[416],total_loss:167.5829|loss_ciou:31.7232|loss_conf:119.9403|loss_cls:15.9195|lr:0.0000 [2021-04-18 21:26:04,282]-[train.py line:231]: === Epoch:[ 0/10],step:[550/562],img_size:[416],total_loss:165.4504|loss_ciou:31.6022|loss_conf:118.1677|loss_cls:15.6805|lr:0.0000 [2021-04-18 21:26:21,893]-[train.py line:231]: === Epoch:[ 0/10],step:[560/562],img_size:[416],total_loss:163.4508|loss_ciou:31.5294|loss_conf:116.4804|loss_cls:15.4410|lr:0.0000

学习率是精度显示问题,可以设置小数点后8位。是否由于没有使用GPU训练导致训练速度过慢

argusswift avatar Apr 19 '21 03:04 argusswift

为什么我训练特别慢,一个epoch要好久,而且学习率为什么显示为0啊

[2021-04-18 21:24:16,560]-[train.py line:231]: === Epoch:[ 0/10],step:[490/562],img_size: [416],total_loss:178.2688|loss_ciou:31.8549|loss_conf:129.4508|loss_cls:16.9630|lr:0.0000 [2021-04-18 21:24:33,861]-[train.py line:231]: === Epoch:[ 0/10],step:[500/562],img_size:[416],total_loss:176.0515|loss_ciou:31.8598|loss_conf:127.4119|loss_cls:16.7799|lr:0.0000 [2021-04-18 21:24:52,436]-[train.py line:231]: === Epoch:[ 0/10],step:[510/562],img_size:[416],total_loss:173.9491|loss_ciou:31.8817|loss_conf:125.4699|loss_cls:16.5975|lr:0.0000 [2021-04-18 21:25:11,434]-[train.py line:231]: === Epoch:[ 0/10],step:[520/562],img_size:[416],total_loss:171.9408|loss_ciou:31.9154|loss_conf:123.6213|loss_cls:16.4042|lr:0.0000 [2021-04-18 21:25:28,948]-[train.py line:231]: === Epoch:[ 0/10],step:[530/562],img_size:[416],total_loss:169.7509|loss_ciou:31.8272|loss_conf:121.7585|loss_cls:16.1653|lr:0.0000 [2021-04-18 21:25:46,376]-[train.py line:231]: === Epoch:[ 0/10],step:[540/562],img_size:[416],total_loss:167.5829|loss_ciou:31.7232|loss_conf:119.9403|loss_cls:15.9195|lr:0.0000 [2021-04-18 21:26:04,282]-[train.py line:231]: === Epoch:[ 0/10],step:[550/562],img_size:[416],total_loss:165.4504|loss_ciou:31.6022|loss_conf:118.1677|loss_cls:15.6805|lr:0.0000 [2021-04-18 21:26:21,893]-[train.py line:231]: === Epoch:[ 0/10],step:[560/562],img_size:[416],total_loss:163.4508|loss_ciou:31.5294|loss_conf:116.4804|loss_cls:15.4410|lr:0.0000

默认训练参数里面使用的是CPU哦~ 要记得切换GPU~

Cimy-wang avatar May 07 '21 12:05 Cimy-wang