• Cjkkkk's avatar
    add data parallel proposal (#1923) · 2de52a89
    Cjkkkk authored
    * add data parallel proposal
    
    * fix mask_weight bug
    
    * add slim pruner support and example
    
    * fix typo
    
    * fix typo
    
    * fix setattr error
    
    * fix buffer update
    
    * rename instrument_layer and prunerLayerWrapper
    
    * fix pylint
    
    * update reverse travsal
    
    * add wrap and unwrap
    
    * add register_buffer API
    
    * update docstring
    
    * update docstring
    
    * add quantizer support
    
    * fix typo
    
    * update MeanActivationPruner, weight_rank_filter_pruner and example
    2de52a89
multi_gpu.py 3.59 KB