linear eval syncbn

configs/benchmarks/linear_classification/imagenet/r50_multihead.py

@@ -17,7 +17,7 @@ model = dict(
         in_indices=[0, 1, 2, 3, 4],
         with_last_layer_unpool=False,
         backbone='resnet50',
-        norm_cfg=dict(type='BN', momentum=0.1, affine=False),
+        norm_cfg=dict(type='SyncBN', momentum=0.1, affine=False),
         num_classes=1000))
 # dataset settings
 data_source_cfg = dict(

configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py

@@ -17,7 +17,7 @@ model = dict(
         in_indices=[0, 1, 2, 3, 4],
         with_last_layer_unpool=False,
         backbone='resnet50',
-        norm_cfg=dict(type='BN', momentum=0.1, affine=False),
+        norm_cfg=dict(type='SyncBN', momentum=0.1, affine=False),
         num_classes=1000))
 # dataset settings
 data_source_cfg = dict(

configs/benchmarks/linear_classification/places205/r50_multihead.py

@@ -17,7 +17,7 @@ model = dict(
         in_indices=[0, 1, 2, 3, 4],
         with_last_layer_unpool=False,
         backbone='resnet50',
-        norm_cfg=dict(type='BN', momentum=0.1, affine=False),
+        norm_cfg=dict(type='SyncBN', momentum=0.1, affine=False),
         num_classes=205))
 # dataset settings
 data_source_cfg = dict(

configs/benchmarks/linear_classification/places205/r50_multihead_sobel.py

@@ -17,7 +17,7 @@ model = dict(
         in_indices=[0, 1, 2, 3, 4],
         with_last_layer_unpool=False,
         backbone='resnet50',
-        norm_cfg=dict(type='BN', momentum=0.1, affine=False),
+        norm_cfg=dict(type='SyncBN', momentum=0.1, affine=False),
         num_classes=205))
 # dataset settings
 data_source_cfg = dict(

configs/selfsup/npid/r50.py

@@ -2,6 +2,7 @@ _base_ = '../../base.py'
 # model settings
 model = dict(
     type='NPID',
+    neg_num=65536,
     pretrained=None,
     backbone=dict(
         type='ResNet',