[Fix] Fix save scheduler state dict with optim wrapper (#375)
* fix save scheduler state dict with optim wrapper * remove for loop and inherit TestParameterScheduler * remove for loop and inherit TestParameterScheduler * minor refine
Showing
- mmengine/optim/optimizer/optimizer_wrapper.py 10 additions, 6 deletionsmmengine/optim/optimizer/optimizer_wrapper.py
- mmengine/optim/scheduler/param_scheduler.py 4 additions, 2 deletionsmmengine/optim/scheduler/param_scheduler.py
- tests/test_optim/test_optimizer/test_optimizer_wrapper.py 0 additions, 1 deletiontests/test_optim/test_optimizer/test_optimizer_wrapper.py
- tests/test_optim/test_scheduler/test_param_scheduler.py 15 additions, 1 deletiontests/test_optim/test_scheduler/test_param_scheduler.py
Loading
Please register or sign in to comment