训练时间太长 - Githubissues

MaybeShewill-CV / lanenet-lane-detection

Unofficial implemention of lanenet model for real time lane detection

Apache License 2.0

2.38k stars 888 forks source link

训练时间太长 #596

Closed gabaebae123 closed 2 months ago

gabaebae123 commented 3 months ago

您好！我使用自制的数据集（图片数为2238张）训练此模型，epoch=300,batch_size=32，2张2080Ti训练一个epoch需要18min，修改batch_size=8之后还是需要很长时间，请问我的设置有什么问题吗？之前只有1500张数据的时候，同样的设置，训练就很快。加了700张数据之后训练速度就慢很多了，慢了差不多10倍，请问为什么呢？

gabaebae123 commented 2 months ago

解决了