From b2f0c075fabdd3d7fa12b76cb5a33740ee66b1e0 Mon Sep 17 00:00:00 2001
From: xiaohangzhan <xiaohangzhan@outlook.com>
Date: Sat, 25 Jul 2020 12:14:47 +0800
Subject: [PATCH] linear eval syncbn

---
 .../benchmarks/linear_classification/imagenet/r50_multihead.py  | 2 +-
 .../linear_classification/imagenet/r50_multihead_sobel.py       | 2 +-
 .../benchmarks/linear_classification/places205/r50_multihead.py | 2 +-
 .../linear_classification/places205/r50_multihead_sobel.py      | 2 +-
 configs/selfsup/npid/r50.py                                     | 1 +
 5 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/configs/benchmarks/linear_classification/imagenet/r50_multihead.py b/configs/benchmarks/linear_classification/imagenet/r50_multihead.py
index 94d3c091..54c995d1 100644
--- a/configs/benchmarks/linear_classification/imagenet/r50_multihead.py
+++ b/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(
diff --git a/configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py b/configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py
index 98b92345..bc3638cd 100644
--- a/configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py
+++ b/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(
diff --git a/configs/benchmarks/linear_classification/places205/r50_multihead.py b/configs/benchmarks/linear_classification/places205/r50_multihead.py
index 2b55a33c..0d1c1a87 100644
--- a/configs/benchmarks/linear_classification/places205/r50_multihead.py
+++ b/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(
diff --git a/configs/benchmarks/linear_classification/places205/r50_multihead_sobel.py b/configs/benchmarks/linear_classification/places205/r50_multihead_sobel.py
index ec8aeb84..2dba622f 100644
--- a/configs/benchmarks/linear_classification/places205/r50_multihead_sobel.py
+++ b/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(
diff --git a/configs/selfsup/npid/r50.py b/configs/selfsup/npid/r50.py
index f6ced74b..3564ca4c 100644
--- a/configs/selfsup/npid/r50.py
+++ b/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',