-
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
ddfb38efJerry 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