• Wenwei Zhang's avatar
    Momentum scheduler (#167) · ba059611
    Wenwei Zhang authored
    * track progress of iter&enum
    
    * restore
    
    * add momentum scheduler
    
    * fix small bug
    
    * cyclic scheduler"
    
    * fix bug
    
    * fix second phase's bug
    
    * reformat
    
    * feature (cosine lr): use relative ratio for more flexible scheduler
    
    * Fix (runner): fix bugs in runner
    
    * Refactor (hook): refactor cosing/cyclic LR/momentum hook with unittest
    
    * Clean unnecessary files and reformat
    
    * Fix memory key error when GPU is not avaliable
    
    * Resolve comments
    
    * Do not print momentum in text log
    
    * Change hook register order
    
    * Refactor max_iter
    
    * Fix max_iter bugs in runner
    
    * Enforce target_ratio to be either tuple or float
    ba059611
test_hooks.py 5.67 KB