• Wenwei Zhang's avatar
    Support to specify LR of DCN's conv_offset (#344) · ed2887bb
    Wenwei Zhang authored
    * Support to specify LR of DCN's conv_offset
    
    * Resolve comments & add unit test
    
    * Resolve formats
    
    * Fix CI for DCN
    
    * Mock DCN when cpu only
    
    * Use mock for cpu testing
    
    * Fix docstring and support ModulatedDCN
    
    * set offset_lr_mult as dcn's arguments, link CU-49u01p
    
    * fix lr bug
    
    * fall back to set LR in constructor
    
    * resolve comments
    ed2887bb
test_optimizer.py 23.3 KB