open-mmlab / mmpose

OpenMMLab Pose Estimation Toolbox and Benchmark.
https://mmpose.readthedocs.io/en/latest/
Apache License 2.0
5.26k stars 1.17k forks source link

fix: handle resume on RTMOModeSwitchHook and YOLOXPoseModeSwitchHook #3045

Open drazicmartin opened 1 month ago

drazicmartin commented 1 month ago

There was an issue when resuming training on theses Hooks (RTMOModeSwitchHook, YOLOXPoseModeSwitchHook). These hooks were not triggered when resuming training because they required the training to go over each specific epoch.

Motivation

Fix the issue

Modification

Add logic to handle the resume, It now applies the switch mode in order until it reach the resumed epoch.

Checklist

Before PR:

After PR: