@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?
@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?