• Harry's avatar
    Add CosineRestartLrUpdaterHook (#319) · f28a7c7e
    Harry authored
    * feat: add CosineRestartLrUpdaterHook
    
    * style: rename period to periods
    
    * fix: bug in period 0
    
    * feat: rename eta_min to min_lr and add min_lr_ratio
    
    * docs: fix docstring of restart lr updater
    
    * refactor: use annealing_cos
    
    * docs: add docstring to annealing_cos
    
    * feat: cosine restart lr update hook
    
    * refactor: modify code order for unittest
    f28a7c7e
test_hooks.py 8.76 KB