search

bubbliiiing / yolov4-pytorch

这是一个YoloV4-pytorch的源码，可以用于训练自己的模型。
MIT License
2.11k stars 611 forks source link

sgd和adam优化器 #314

Open 77serendipity opened 2 years ago

77serendipity commented 2 years ago

将sgd优化器换成adam优化器，训练速度还是很慢，电脑没有显卡，用的核显。

image

bubbliiiing commented 2 years ago

adam是优化的速度快，不是这个显示的速度快。反传的时间都是一样的。

77serendipity commented 2 years ago

之前用过v5训练速度挺快的。用您这套训练660张图片380epoch，一个epoch大概10分钟，感觉好慢啊。训练完大概要三天haha

77serendipity commented 2 years ago

您好想问下，这个loss曲线正常嘛，10几个小时之训练了30几个epoch image

bubbliiiing commented 2 years ago

1、v5是s，v5可没有s 2、比较正常

77serendipity commented 2 years ago

好的谢谢，我从voc数据集上找了三类测试模型。一共660张图片，建议的epoch是380左右。我的电脑没有独显（服务器坏了只能用自己的cpu测一下），跑了三天三夜多才跑了108epoch image image 这个训练一个epoch从66到133是学习率自动调整改变的嘛。感谢指导

bubbliiiing commented 2 years ago

学习率是自动调整的呀。

bubbliiiing commented 2 years ago

1、v5是s，v5可没有s 2、比较正常

写错了，是v4没有s

77serendipity commented 2 years ago

谢谢，想问下大佬有没有训练过VOC数据集或者自己的数据集，训练速度如何，还有您用的什么设备，gpu型号。