Wolfwjs / GANet

A Keypoint-based Global Association Network for Lane Detection. Accepted by CVPR 2022
MIT License
239 stars 29 forks source link

abouttraining loss #23

Closed Godaddy-xie closed 2 years ago

Godaddy-xie commented 2 years ago

Can u provide the training log? i have try toI tried to fit on a small dataset (1000 culanes, or my own dataset , batchsize = 1), but focalloss1 has been around 1.35 (culane 800*320) and cannot be decreased. i have let other loss weight zero. image

image

spyflying commented 2 years ago

It is unusual that the focalloss keeps high. I suppose batchsize=1 might be the reason. You can try to use larger batchsize during training. Also, try to use the provided checkpoint for parameter initialization and finetune on your dataset with smaller learing rate. Since the original log is not available now, I trained on tusimple for several epochs and paste the log below. Hope it could help you.

focal

log