New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

loss nan #27

Open

alicera opened this issue Jun 3, 2020 · 2 comments

alicera commented Jun 3, 2020 •

edited

Loading

python main.py
I dont change the setting to train

4 |ETA: 0:05:49 |loss nan |hm_loss nan |lm_loss nan |wh_loss nan |off_loss nanmulti_pose/dla |########### | train: [8][546/1583]|Tot: 0:03:05 |ETA: 0:05:49 |loss nan |hm_loss nan |lm_loss nan |wh_loss nan |off_lo

how to resize your images size for training

alicera changed the title ~~No module named 'external.nms'~~ loss nan

iytChen commented Jul 5, 2020

same question， anyone knows how to sovle this problem?

YangYangGirl commented Sep 26, 2020

CenterFace.pytorch/src/lib/trains/multi_pose.py

Line 34 in 37a351b

# if opt.hm_hp and not opt.mse_loss:

change to this:
if not opt.mse_loss:
output['hm'] = _sigmoid(output['hm'])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment