search

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
7.05k stars 1.34k forks source link

Pytorch training on 2D face alignment always fails to converge #83

Closed tyshiwo closed 6 years ago

tyshiwo commented 6 years ago

Hi,

Recently, I re-implemented the training code of your heatmap model in pytorch since your released code is for inference only. We used Adam/RMSProp as the optimizer, and 300W as the training set. However, we failed the let the model converge and the loss cannot decrease as long as it reaches 0.006.

We checked the input image and the ground truth coordinates, which are both correct. I wonder if your pytorch training code is available or any suggestions to let the pytorch training work are welcome.

Thanks!

shayxurui commented 6 years ago

hello: i wanna to retrain this network,too. so could u share ur training code? thks,my email is 17212010044@fudan.edu.cn

1adrianb commented 6 years ago

Unfortunatelly we have no imediate plans on preparing the pytorch training code, however if you correctly converted the lua training code you should be able to match or get close to the reported results.