Model not getting trained

[IAM-P-LP]

Hi,
Thanks for this wonderful repo.
While training the model with this command 'python3 train.py --phi 0 --weights /home/priyalakshmip/pose_est_efficientpose/EfficientPose/COCO/efficientdet-d0.h5 linemod /home/priyalakshmip/pose_est_efficientpose/Linemod_preprocessed/ --object-id 4'
I am receiving nan in loss and the map is not also improving.I saw the issue #30.
I have installed cudatoolkit=10.0 and I am using the pretrained weight of efficientdet as well.Can you please suggest why the model is not learning and the steps to solve it?

N.B : Saw this warning

Is that the reason why the model isnt learning?

[DavidJimenezS]

No sure if you were able to solve the problem. But in short is due to GPU compatibility with cuda toolkit look here:

#69 (comment)