fixed parrots SyncBN

open-mmlab · Jun 25, 2021 · d9af6b0 · d9af6b0
1 parent 9838447
commit d9af6b0
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/mmcv/utils/parrots_wrapper.py b/mmcv/utils/parrots_wrapper.py
@@ -82,10 +82,6 @@ def _get_norm():
 
 class SyncBatchNorm(SyncBatchNorm_):
 
-    def _specify_ddp_gpu_num(self, gpu_size):
-        if TORCH_VERSION != 'parrots':
-            super()._specify_ddp_gpu_num(gpu_size)
-
     def _check_input_dim(self, input):
         if TORCH_VERSION == 'parrots':
             if input.dim() < 2: