mxnet_Realtime_Multi-Person_Pose_Estimation
mxnet_Realtime_Multi-Person_Pose_Estimation copied to clipboard
Load_checkpoint did not work well
@dragonfly90 thanks for your share~ I'm try to modify it to fit AI challenge keypoint detection. Now i found a problem about load_checkpoint and save_checkpoint function. such as https://github.com/dragonfly90/mxnet_Realtime_Multi-Person_Pose_Estimation/blob/master/TrainWeight.py#L89 For example, i train 10 images every epoch, after 5 epoches,I get paf error and heat error about 20, and i save the params as "xxx-0005.params", but i got the two error above 200 when i try to use the model parameters with load_checkpoint("xxx",5) to continue train the model. And i met the same problem when I trained 210000 datas after 3 epoches. Do u know why this happened?
@AndyWolfZwei are they in the same directory?