diff --git a/docs/MODEL_ZOO.md b/docs/MODEL_ZOO.md index 492fe291..0b01f55d 100644 --- a/docs/MODEL_ZOO.md +++ b/docs/MODEL_ZOO.md @@ -9,24 +9,25 @@ Since we don't have sufficient GPUs to run these large-scale experiments, your c * Experiments with the same batch size are directly comparable in speed. * The table records the implementors who implemented the methods (either by themselves or refactoring from other repos), and the experimenters who performed experiments and reproduced the results. The experimenters should be responsible for the evaluation results on all the benchmarks, and the implementors should be responsible for the implementation as well as the results; If the experimenter is not indicated, an implementator is the experimenter by default. -
Method (Implementator) | Config (Experimenter) | Remarks | Download link | Batch size | Epochs | Time per epoch |
---|---|---|---|---|---|---|
ImageNet | - | torchvision | imagenet_r50-21352794.pth | - | - | - |
Random | - | kaiming | random_r50-5d0fa71b.pth | - | - | - |
Relative-Loc (@Jiahao000) | selfsup/relative_loc/r50.py | default | relative_loc_r50-342c9097.pth | 512 | 70 | 21min17s |
Rotation-Pred (@XiaohangZhan) | selfsup/rotation_pred/r50.py | default | rotation_r50-cfab8ebb.pth | 128 | 70 | 49min58s |
DeepCluster (@XiaohangZhan) | selfsup/deepcluster/r50.py | default | deepcluster_r50-bb8681e2.pth | 512 | 200 | 41min57s |
NPID (@XiaohangZhan) | selfsup/npid/r50.py | default | npid_r50-dec3df0c.pth | 256 | 200 | 20min5s |
selfsup/npid/r50_ensure_neg.py | ensure_neg=True | npid_r50_ensure_neg-ce09b7ae.pth | ||||
ODC (@XiaohangZhan) | selfsup/odc/r50_v1.py (@Jiahao000) | default | odc_r50_v1-5af5dd0c.pth | 512 | 440 | 28min22s |
MoCo (@XiaohangZhan) | selfsup/moco/r50_v1.py | default | moco_r50_v1-4ad89b5c.pth | 256 | 200 | 22min58s |
MoCo v2 (@XiaohangZhan) | selfsup/moco/r50_v2.py | default | moco_r50_v2-e3b0c442.pth | 256 | 200 | 25min7s |
SimCLR (@XiaohangZhan) | selfsup/simclr/r50_bs256_ep200.py | default | simclr_r50_bs256_ep200-4577e9a6.pth | 256 | 200 | 32min13s |
selfsup/simclr/r50_bs256_ep200_mocov2_neck.py | -> MoCo v2 neck | simclr_r50_bs256_ep200_mocov2_neck-0d6e5ff2.pth | ||||
BYOL (@XiaohangZhan) | selfsup/byol/r50_bs4096_ep200.py (@xieenze) | default | byol_r50-e3b0c442.pth | 4096 (256 for speed test) | 200 | 30min57s |
selfsup/byol/r50_bs256_accumulate16_ep300.py (@scnuhealthy) | default | byol_r50_bs256_accmulate16_ep300-5df46722.pth | 256 | 300 | ||
selfsup/byol/r50_bs2048_accumulate2_ep200_fp16.py (@xieenze) | default | byol_r50_bs2048_accmulate2_ep200-e3b0c442.pth | 256 | 200 |
Method (Implementator) | Config (Experimenter) | Remarks | Download link | Batch size | Epochs |
---|---|---|---|---|---|
ImageNet | - | torchvision | imagenet_r50-21352794.pth | - | - |
Random | - | kaiming | random_r50-5d0fa71b.pth | - | - |
Relative-Loc (@Jiahao000) | selfsup/relative_loc/r50.py | default | relative_loc_r50-342c9097.pth | 512 | 70 |
Rotation-Pred (@XiaohangZhan) | selfsup/rotation_pred/r50.py | default | rotation_r50-cfab8ebb.pth | 128 | 70 |
DeepCluster (@XiaohangZhan) | selfsup/deepcluster/r50.py | default | deepcluster_r50-bb8681e2.pth | 512 | 200 |
NPID (@XiaohangZhan) | selfsup/npid/r50.py | default | npid_r50-dec3df0c.pth | 256 | 200 |
selfsup/npid/r50_ensure_neg.py | ensure_neg=True | npid_r50_ensure_neg-ce09b7ae.pth | |||
ODC (@XiaohangZhan) | selfsup/odc/r50_v1.py (@Jiahao000) | default | odc_r50_v1-5af5dd0c.pth | 512 | 440 |
MoCo (@XiaohangZhan) | selfsup/moco/r50_v1.py | default | moco_r50_v1-4ad89b5c.pth | 256 | 200 |
MoCo v2 (@XiaohangZhan) | selfsup/moco/r50_v2.py | default | moco_r50_v2-e3b0c442.pth | 256 | 200 |
SimCLR (@XiaohangZhan) | selfsup/simclr/r50_bs256_ep200.py | default | simclr_r50_bs256_ep200-4577e9a6.pth | 256 | 200 |
selfsup/simclr/r50_bs256_ep200_mocov2_neck.py | -> MoCo v2 neck | simclr_r50_bs256_ep200_mocov2_neck-0d6e5ff2.pth | |||
BYOL (@XiaohangZhan) | selfsup/byol/r50_bs4096_ep200.py (@xieenze) | default | byol_r50-e3b0c442.pth | 4096 (256 for speed test) | 200 |
selfsup/byol/r50_bs256_accumulate16_ep300.py (@scnuhealthy) | default | byol_r50_bs256_accmulate16_ep300-5df46722.pth | 256 | 300 | |
selfsup/byol/r50_bs2048_accumulate2_ep200_fp16.py (@xieenze) | default | byol_r50_bs2048_accmulate2_ep200-e3b0c442.pth | 256 | 200 |