Skip to content
Snippets Groups Projects
Commit ddfb38ef authored by Jerry XU's avatar Jerry XU Committed by Kai Chen
Browse files

add pytorch 1.1.0 SyncBN support (#577) 

* 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
parent c52cdd62
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment