• Gu Wang's avatar
    [Feature] Add flat cosine lr updater (#1066) · 025ee9f5
    Gu Wang authored
    * add flat cosine lr updater
    
    * add test
    
    * add doc
    
    * update doc
    
    * reformat
    
    * update unittest
    
    * update test flat cos
    
    * remove momentum hook test
    
    * update test
    
    * change assert to ValueError
    
    * fix unittest
    
    * add by_epoch=True unittest
    
    * change to start_percent
    
    * change to start_percent in test
    025ee9f5
test_hooks.py 41.5 KB