loss nan

[alicera]

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

[iytChen]

same question， anyone knows how to sovle this problem?

[YangYangGirl]

https://github.com/chenjun2hao/CenterFace.pytorch/blob/37a351be07606136561bc56eae716f8575b61e00/src/lib/trains/multi_pose.py#L34

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