diff --git a/docs/MODEL_ZOO.md b/docs/MODEL_ZOO.md
index 889116ef..abcfe013 100644
--- a/docs/MODEL_ZOO.md
+++ b/docs/MODEL_ZOO.md
@@ -127,7 +127,7 @@ Coming soon.
* When training with 1% ImageNet, we find hyper-parameters especially the learning rate greatly influence the performance. Hence, we prepare a list of settings with the base learning rate from \{0.001, 0.01, 0.1\} and the learning rate multiplier for the head from \{1, 10, 100\}. We choose the best performing setting for each method.
* Please use `--deterministic` in this benchmark.
-
Method | Config | Remarks | Optimal setting for ImageNet 1% | ImageNet 1% |
---|
top-1 | top-5 |
ImageNet | - | torchvision | r50_lr0_001_head100.py | 68.68 | 88.87 |
Random | - | kaiming | r50_lr0_01_head1.py | 1.56 | 4.99 |
Relative-Loc | selfsup/relative_loc/r50.py | default | | | |
Rotation-Pred | selfsup/rotation_pred/r50.py | default | r50_lr0_01_head100.py | 18.98 | 44.05 |
DeepCluster | selfsup/deepcluster/r50.py | default | r50_lr0_01_head1_sobel.py | 33.44 | 58.62 |
NPID | selfsup/npid/r50.py | default | r50_lr0_01_head100.py | 27.95 | 54.37 |
ODC | selfsup/odc/r50_v1.py | default | r50_lr0_1_head100.py | 32.39 | 61.02 |
MoCo | selfsup/moco/r50_v1.py | default | r50_lr0_01_head100.py | 33.15 | 61.30 |
MoCo v2 | selfsup/moco/r50_v2.py | default | r50_lr0_01_head100.py | 38.71 | 67.90 |
| selfsup/moco/r50_v2_simclr_neck.py | -> SimCLR neck
| | | |
SimCLR | selfsup/simclr/r50_bs256_ep200.py | default | r50_lr0_01_head100.py | 36.09 | 64.50 |
| selfsup/simclr/r50_bs256_ep200_mocov2_neck.py | -> MoCo v2 neck | | | |
BYOL | selfsup/byol/r50.py | default | | | |
+Method | Config | Remarks | Optimal setting for ImageNet 1% | ImageNet 1% |
---|
top-1 | top-5 |
ImageNet | - | torchvision | r50_lr0_001_head100.py | 68.68 | 88.87 |
Random | - | kaiming | r50_lr0_01_head1.py | 1.56 | 4.99 |
Relative-Loc | selfsup/relative_loc/r50.py | default | | | |
Rotation-Pred | selfsup/rotation_pred/r50.py | default | r50_lr0_01_head100.py | 18.98 | 44.05 |
DeepCluster | selfsup/deepcluster/r50.py | default | r50_lr0_01_head1_sobel.py | 33.44 | 58.62 |
NPID | selfsup/npid/r50.py | default | r50_lr0_01_head100.py | 27.95 | 54.37 |
ODC | selfsup/odc/r50_v1.py | default | r50_lr0_1_head100.py | 32.39 | 61.02 |
MoCo | selfsup/moco/r50_v1.py | default | r50_lr0_01_head100.py | 33.15 | 61.30 |
MoCo v2 | selfsup/moco/r50_v2.py | default | r50_lr0_01_head100.py | 38.71 | 67.90 |
| selfsup/moco/r50_v2_simclr_neck.py | -> SimCLR neck
| r50_lr0_01_head100.py | 31.37
| 59.65 |
SimCLR | selfsup/simclr/r50_bs256_ep200.py | default | r50_lr0_01_head100.py | 36.09 | 64.50 |
| selfsup/simclr/r50_bs256_ep200_mocov2_neck.py | -> MoCo v2 neck | r50_lr0_01_head100.py | 36.31 | 64.68 |
BYOL | selfsup/byol/r50.py | default | | | |
Method | Config | Remarks | Optimal setting for ImageNet 1% | ImageNet 1% |
---|
top-1 | top-5 |
ImageNet | - | torchvision | r50_lr0_01_head1.py | 63.10 | 85.73 |
Random | - | kaiming | r50_lr0_01_head1.py | 1.56 | 4.99 |
Relative-Loc | selfsup/relative_loc/r50.py | default | | | |
Rotation-Pred | selfsup/rotation_pred/r50.py | default | r50_lr0_01_head100.py | 18.98 | 44.05 |
DeepCluster | selfsup/deepcluster/r50.py | default | r50_lr0_01_head1_sobel.py | 33.44 | 58.62 |
NPID | selfsup/npid/r50.py | default | r50_lr0_01_head100.py | 27.95 | 54.37 |
ODC | selfsup/odc/r50_v1.py | default | r50_lr0_1_head100.py | 32.39 | 61.02 |
MoCo | selfsup/moco/r50_v1.py | default | r50_lr0_01_head100.py | 33.15 | 61.30 |
MoCo v2 | selfsup/moco/r50_v2.py | default | r50_lr0_01_head100.py | 38.71 | 67.90 |
| selfsup/moco/r50_v2_simclr_neck.py | -> SimCLR neck
| | | |
SimCLR | selfsup/simclr/r50_bs256_ep200.py | default | r50_lr0_01_head100.py | 36.09 | 64.50 |
| selfsup/simclr/r50_bs256_ep200_mocov2_neck.py | -> MoCo v2 neck | | | |
BYOL | selfsup/byol/r50.py | default | | | |