关于训练时间!

[m-wei]

请问从头开始训练，大概需要多长时间，loss一般能达到多少呢

[dvlee1024]

看你机器的算力和图集大小，昨晚用WIDER FACE的10000张图片训练人脸，1080显卡，batchsize 8，一小时收敛到20~80的loss，后面持续了几个小时，loss跳动幅度特别大。训练了一晚上，后来loss还变成nan了。

我拿epoch20~30来测试，识别人脸效果还可以。

[m-wei]

@dvlee1024， 你的batchsize取的是多大的呢

[dvlee1024]

@dvlee1024， 你的batchsize取的是多大的呢

batchsize 8。 我现在重新训练，数据集还是用WIDER FACE，把小脸、模糊等anno去掉了，把IMG_AUG关了，收敛效果比之前好多了，训练十分钟loss在15左右

十分钟后，loss下降到10以内，然后就出现nan了， @YunYang1994 ，是不是有bug啊

[m-wei]

@dvlee1024，那还可以的呀，我是默认的batchsize:2，我训了大概16小时，目前在1-10范围跳动，我待会测试一下

[m-wei]

@dvlee1024，请问下这个测试代码是哪个，我训练出来的模型yolov3.data-00000-of-00001这个，image_demo里面是yolo3.weight，这个是怎么转的或者其他方式测试吗，谢谢！

[orange-dragon]

@dvlee1024，请问下这个测试代码是哪个，我训练出来的模型yolov3.data-00000-of-00001这个，image_demo里面是yolo3.weight，这个是怎么转的或者其他方式测试吗，谢谢！

请问这个问题解决了吗？方便介绍一下是怎么转换成.weight的吗？

[ho1messi]

@dvlee1024，请问下这个测试代码是哪个，我训练出来的模型yolov3.data-00000-of-00001这个，image_demo里面是yolo3.weight，这个是怎么转的或者其他方式测试吗，谢谢！

请问这个问题解决了吗？方便介绍一下是怎么转换成.weight的吗？

test.py 48 行有个 model.load_weights("./yolov3") 可以直接加载训练保存的权重