How to train pre-train model

thohemp / 6DRepNet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.

MIT License

550 stars 72 forks source link

How to train pre-train model #20

Closed BenjaminGit001 closed 2 years ago

BenjaminGit001 commented 2 years ago

Hi, I tried to train the model from scratch, seems hard to train comparable performance as training model from the pre-train model. My question is how to train the pre-train model or how to train the similar performance from scratch?

Thanks!

thohemp commented 2 years ago

I updated the default training settings. This should help you to achieve the reported results more easily (or in my trainings even surpass them). Let me know if you still struggle after using them.