NVlabs / CenterPose

Single-Stage Keypoint-based Category-level Object Pose Estimation from an RGB Image (ICRA 2022)
Other
285 stars 34 forks source link

Saving model multiple times #3

Closed MJ1998 closed 2 years ago

MJ1998 commented 2 years ago

https://github.com/NVlabs/CenterPose/blob/a92d93b09660e266a43533a0dc722445873829ea/src/main_CenterPose.py#L104

Uio96 commented 2 years ago

It is designed to save the latest checkpoint during the process. So it will save the weight multiple times.