Skip to content
  • Jerry XU's avatar
    ddfb38ef
    add pytorch 1.1.0 SyncBN support (#577) · ddfb38ef
    Jerry XU authored
    * add pytorch 1.1.0 SyncBN support
    
    * change BatchNorm2d to _BatchNorm and call freeze after train
    
    * add freeze back to init function
    
    * fixed indentation typo in adding freeze
    
    * use SyncBN protect member func to set ddp_gpu_num
    
    * Update README.md
    
    update pytorch version to 1.1
    ddfb38ef
    add pytorch 1.1.0 SyncBN support (#577)
    Jerry XU authored
    * add pytorch 1.1.0 SyncBN support
    
    * change BatchNorm2d to _BatchNorm and call freeze after train
    
    * add freeze back to init function
    
    * fixed indentation typo in adding freeze
    
    * use SyncBN protect member func to set ddp_gpu_num
    
    * Update README.md
    
    update pytorch version to 1.1
Loading