diff --git a/docs/MODEL_ZOO.md b/docs/MODEL_ZOO.md
index f47de241..da8f1e04 100644
--- a/docs/MODEL_ZOO.md
+++ b/docs/MODEL_ZOO.md
@@ -13,10 +13,11 @@
### ImageNet Linear Classification
-
Method | Config | Remarks | ImageNet (Multi) | ImageNet (Last) |
---|
feat1 | feat2 | feat3 | feat4 | feat5 | avgpool |
ImageNet | - | torchvision | 15.18 | 33.96 | 47.86 | 67.56 | 76.17 | 74.12 |
Random | - | kaiming | | | | | | 4.35 |
Relative-Loc | selfsup/relative_loc/r50.py | default | | | | | | |
Rotation-Pred | selfsup/rotation_pred/r50.py | default | 12.89 | 34.30 | 44.91 | 54.99 | 49.09 | |
DeepCluster | selfsup/deepcluster/r50.py | default | | | | | | 46.92 |
NPID | selfsup/npid/r50.py | default | 14.28 | 31.20 | 40.68 | 54.46 | 56.61 | |
ODC | selfsup/odc/r50_v1.py | default | 14.75 | 31.55 | 42.49 | 55.72 | 57.57 | |
MoCo | selfsup/moco/r50_v1.py | default | 15.32 | 33.08 | 44.68 | 57.27 | 60.60 | 61.02 |
MoCo v2 | selfsup/moco/r50_v2.py | default | 15.35 | 34.57 | 45.81 | 60.96 | 66.72 | 67.02 |
| selfsup/moco/r50_v2.py | -> SimCLR neck
| | | | | | |
SimCLR | selfsup/simclr/r50_bs256.py | default | | | | | | |
BYOL | selfsup/byol/r50.py | default | | | | | | |
+Method | Config | Remarks | ImageNet (Multi) | ImageNet (Last) |
---|
feat1 | feat2 | feat3 | feat4 | feat5 | avgpool |
ImageNet | - | torchvision | 15.18 | 33.96 | 47.86 | 67.56 | 76.17 | 74.12 |
Random | - | kaiming | | | | | | 4.35 |
Relative-Loc | selfsup/relative_loc/r50.py | default | | | | | | |
Rotation-Pred | selfsup/rotation_pred/r50.py | default | 12.89 | 34.30 | 44.91 | 54.99 | 49.09 | |
DeepCluster | selfsup/deepcluster/r50.py | default | | | | | | 46.92 |
NPID | selfsup/npid/r50.py | default | 14.28 | 31.20 | 40.68 | 54.46 | 56.61 | |
ODC | selfsup/odc/r50_v1.py | default | 14.75 | 31.55 | 42.49 | 55.72 | 57.57 | |
MoCo | selfsup/moco/r50_v1.py | default | 15.32 | 33.08 | 44.68 | 57.27 | 60.60 | 61.02 |
MoCo v2 | selfsup/moco/r50_v2.py | default | 15.35 | 34.57 | 45.81 | 60.96 | 66.72 | 67.02 |
| selfsup/moco/r50_v2.py | -> SimCLR neck
| | | | | | |
SimCLR | selfsup/simclr/r50_bs256_ep200.py | default | | | | | | |
| selfsup/simclr/r50_bs256_ep200_mocov2_neck.py | -> MoCo v2 neck | | | | | | |
BYOL | selfsup/byol/r50.py | default | | | | | | |
### Place Linear Classification
+Coming soon.
### ImageNet Semi-Supervised Classification
@@ -28,3 +29,5 @@
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.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 | | | |
### PASCAL VOC07+12 Object Detection
+
+Coming soon.