the train loss is nan - Githubissues

wangziren1 / pytorch_pose_proposal_networks

Pytorch implementation of pose proposal networks

MIT License

105 stars 23 forks source link

the train loss is nan #4

Open MrYANG23 opened 5 years ago

MrYANG23 commented 5 years ago

hi! thanks for your work ,when i train this repo i get nan loss no matter use the resnet18 or using the resnet50. can you give me some advices,thank you !

PuAnysh commented 5 years ago

Me Too.this problem will occur while i use loss_coor in final loss.

wangziren1 commented 5 years ago

Maybe, dropping learning rate by ten times or five times helps.