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).
MethodConfigRemarksDownload linkBatch sizeEpochsTime per epoch
ImageNet-torchvisionimagenet_r50-21352794.pth---
Random-kaimingrandom_r50-5d0fa71b.pth---
Relative-Locselfsup/relative_loc/r50.pydefaultrelative_loc_r50-342c9097.pth5127021min17s
Rotation-Predselfsup/rotation_pred/r50.pydefaultrotation_r50-cfab8ebb.pth1287049min58s
DeepClusterselfsup/deepcluster/r50.pydefaultdeepcluster_r50-bb8681e2.pth51220041min57s
NPIDselfsup/npid/r50.pydefaultnpid_r50-dec3df0c.pth25620020min5s
ODCselfsup/odc/r50_v1.pydefaultodc_r50_v1-5af5dd0c.pth51244028min22s
MoCoselfsup/moco/r50_v1.pydefaultmoco_r50_v1-4ad89b5c.pth25620022min58s
MoCo v2selfsup/moco/r50_v2.pydefaultmoco_r50_v2-58f10cfe.pth25620055min43s
selfsup/moco/r50_v2_simclr_neck.py-> SimCLR neck
moco_r50_v2_simclr_neck-70379356.pth
SimCLRselfsup/simclr/r50_bs256_ep200.pydefaultsimclr_r50_bs256_ep200-4577e9a6.pth2562001h1min7s
selfsup/simclr/r50_bs256_ep200_mocov2_neck.py-> MoCo v2 necksimclr_r50_bs256_ep200_mocov2_neck-0d6e5ff2.pth
BYOLselfsup/byol/r50_bs4096_ep200.pydefaultbyol_r50-e3b0c442.pth409620014min40s