EfficientPose icon indicating copy to clipboard operation
EfficientPose copied to clipboard

Published 20 hours ago • verified publisher icon ybkscht

Model not getting trained

Open IAM-P-LP opened this issue 1 year ago • 1 comments

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 https://github.com/ybkscht/EfficientPose/issues/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? image

N.B : Saw this warning image Is that the reason why the model isnt learning?

IAM-P-LP avatar Aug 25 '23 11:08 IAM-P-LP

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

https://github.com/ybkscht/EfficientPose/issues/69#issuecomment-1910654889

DavidJimenezS avatar Jan 30 '24 03:01 DavidJimenezS