diff --git a/docs/MODEL_ZOO.md b/docs/MODEL_ZOO.md index 30a1e5f0..c99e2e13 100644 --- a/docs/MODEL_ZOO.md +++ b/docs/MODEL_ZOO.md @@ -1,6 +1,6 @@ # Model Zoo -## Pre-trained model download links +## Pre-trained model download links and speed test (The GPUs are NVIDIA Tesla V100).
Method | Config | Remarks | Download link | Batch size | Epochs | Time per epoch |
---|---|---|---|---|---|---|
ImageNet | - | torchvision | imagenet_r50-21352794.pth | - | - | - |
Random | - | kaiming | random_r50-5d0fa71b.pth | - | - | - |
Relative-Loc | selfsup/relative_loc/r50.py | default | relative_loc_r50-342c9097.pth | 512 | 70 | 21min17s |
Rotation-Pred | selfsup/rotation_pred/r50.py | default | rotation_r50-cfab8ebb.pth | 128 | 70 | 49min58s |
DeepCluster | selfsup/deepcluster/r50.py | default | deepcluster_r50-bb8681e2.pth | 512 | 200 | 41min57s |
NPID | selfsup/npid/r50.py | default | npid_r50-dec3df0c.pth | 256 | 200 | 20min5s |
ODC | selfsup/odc/r50_v1.py | default | odc_r50_v1-5af5dd0c.pth | 512 | 440 | 28min22s |
MoCo | selfsup/moco/r50_v1.py | default | moco_r50_v1-4ad89b5c.pth | 256 | 200 | 22min58s |
MoCo v2 | selfsup/moco/r50_v2.py | default | moco_r50_v2-58f10cfe.pth | 256 | 200 | 55min43s |
selfsup/moco/r50_v2_simclr_neck.py | -> SimCLR neck | moco_r50_v2_simclr_neck-70379356.pth | ||||
SimCLR | selfsup/simclr/r50_bs256_ep200.py | default | simclr_r50_bs256_ep200-4577e9a6.pth | 256 | 200 | 1h1min7s |
selfsup/simclr/r50_bs256_ep200_mocov2_neck.py | -> MoCo v2 neck | simclr_r50_bs256_ep200_mocov2_neck-0d6e5ff2.pth | ||||
BYOL | selfsup/byol/r50_bs4096_ep200.py | default | byol_r50-e3b0c442.pth | 4096 | 200 | 14min40s |