search

vacancy / Synchronized-BatchNorm-PyTorch

Synchronized Batch Normalization implementation in PyTorch.
MIT License
1.5k stars 189 forks source link

compatible with pytorch 1.4.0 #36

Closed wang93 closed 4 years ago

wang93 commented 4 years ago
  1. add 'track_running_stats' parameter when initting objects of SynchronizedBatchNorm classes.
  2. delete 'super(SynchronizedBatchNorm2d, self)._check_input_dim(input)' in the '_check_input_dim' functions, as it may inccur 'NotImplement Error'.
vacancy commented 4 years ago

Thanks for your contribution! The patch has been merged.