Hugging Face's logo

erow
/
SSL

Model card Files
xet
Community
gent commited on
Commit
74f3dc1
·
1 Parent(s): a4f8222

refactor keys and add vit

Browse files
Files changed (15) hide show
  1. download.sh +31 -0
  2. resnet50/dino_resnet50_pretrain.pth +2 -2
  3. resnet50/gpaco_resnet50_IN1K.pth +2 -2
  4. resnet50/mocov3_resnet50_e1000.pth +2 -2
  5. resnet50/simclr_resnet50_e1000.pth +2 -2
  6. resnet50/supcon_e600_resnet50.pth +2 -2
  7. resnet50/mocov2_resnet50_e800.pth → vit/ibot_vitb_in1k.pth +2 -2
  8. vit/ibot_vitb_in1krand.pth +3 -0
  9. vit/ibot_vitl_in1krand.pth +3 -0
  10. vit/ibot_vits_in1k.pth +3 -0
  11. vit/mcssl_vitb_in1k.pth +3 -0
  12. vit/moco_vitb_in1k.pth +3 -0
  13. vit/moco_vits_in1k.pth +3 -0
  14. vit/sit_vitb_in1k.pth +3 -0
  15. vit/sit_vits_in1k.pth +3 -0
download.sh ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ # /users/jw02425/models/SSL/vit/download.sh
3
+ # Download iBOT checkpoints and save as: ibot_{arch}_in1k(.pth)
4
+ set -euo pipefail
5
+
6
+ OUT_DIR=${1:-.}
7
+ mkdir -p "$OUT_DIR"
8
+
9
+ declare -A urls
10
+ urls["ibot_vits_in1k"]="https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/archive/2022/ibot/vits_16/checkpoint_teacher.pth"
11
+ urls["ibot_vitb_in1k"]="https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/archive/2022/ibot/vitb_16/checkpoint_teacher.pth"
12
+ # urls["ibot_vitl_in1k"]="https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/archive/2022/ibot/vitl_16/checkpoint_teacher.pth"
13
+ urls["ibot_vitb_in1krand"]="https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/archive/2022/ibot/vitb_16_rand_mask/checkpoint_teacher.pth"
14
+ # urls["ibot_vitl_in1krand"]="https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/archive/2022/ibot/vitl_16_rand_mask/checkpoint_teacher.pth"
15
+
16
+ # moco
17
+ urls["moco_vits_in1k"]="https://dl.fbaipublicfiles.com/moco-v3/vit-s-300ep/vit-s-300ep.pth.tar"
18
+ urls["moco_vitb_in1k"]="https://dl.fbaipublicfiles.com/moco-v3/vit-b-300ep/vit-b-300ep.pth.tar"
19
+
20
+
21
+
22
+ for name in "${!urls[@]}"; do
23
+ url="${urls[$name]}"
24
+ out_path="$OUT_DIR/${name}.pth"
25
+ if [[ ! -f "$out_path" ]]; then
26
+ echo "File $out_path already exists, skipping download."
27
+ continue
28
+ fi
29
+ echo "Downloading $url -> $out_path"
30
+ wget -c -O "$out_path" "$url"
31
+ done
resnet50/dino_resnet50_pretrain.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:156f8c4166a23dc2951ae811e39d76a06269c565932edf647c0187e65cd7aa7c
3
- size 94345885
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:960937c3a218fd139e19ad50235dab1c126e329220475ef10fbfe19bf37ecf75
3
+ size 94346918
resnet50/gpaco_resnet50_IN1K.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ea827ba2aa33f9b6c833948f73b2df27f633dd7bbcff3435a9eea2afa401cf84
3
- size 141422012
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81a21f01531d4e0d9b11ed0bf5f416dedd3db59b6ef52dd8352f1bdacd4af5b0
3
+ size 94342990
resnet50/mocov3_resnet50_e1000.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e6761f0cf2bf7f876cf6d374a5d4ffbabdd7a51159dbfcd8649b8cb6e1b3c498
3
- size 132162954
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c89db04468fecd2b9a741e676480f7113cb449cd3bee113bce5e205558675ef
3
+ size 94343634
resnet50/simclr_resnet50_e1000.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d57ce245acde0ed5b5fd510581a26d456aaf77d3b3345c2307bac60976269375
3
- size 94343634
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9be5042dbba62a826f2c9dbef2699cff79b588ad5feff063cd498bf63c07281a
3
+ size 94346596
resnet50/supcon_e600_resnet50.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:828931864685ad197aa9a90da9f69a676a1eaac4c81d09a950ef69c9f3a6e160
3
- size 94339448
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:653d2216e9b6a1acd1431619e54074aaf9099ea69a77ab717dddd3f60e7fa07b
3
+ size 94346274
resnet50/mocov2_resnet50_e800.pth → vit/ibot_vitb_in1k.pth RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:323a0889527728205490ba224efc32ede629fb92c11277285067164b14257879
3
- size 112184508
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58a24971d2d0d380ff54163441cdff3b03388ecc7522007ffbac0b66b7482b20
3
+ size 343241894
vit/ibot_vitb_in1krand.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:725593a0974f5e1eb0f7066f16d14a7f5b4e2946c02541c0e0f75e3614035cbb
3
+ size 343242510
vit/ibot_vitl_in1krand.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83e891996d57f2b67ddfa0d35fdf1bcd56b3ab333c0de6721d7af558dd8570af
3
+ size 1213300462
vit/ibot_vits_in1k.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:200991eaa88719ce50c2adffa2db603551d1214e3709e8478011e205c21d7954
3
+ size 86709926
vit/mcssl_vitb_in1k.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32ff4855b600aff6942ee7ca7ea9dcda13b9d0aa7114e021ef014ec099025da9
3
+ size 343242048
vit/moco_vitb_in1k.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:842f237cebd0bd3594cf5849aac1dc4c713357be231fcef3064a9a236e08ae91
3
+ size 343241894
vit/moco_vits_in1k.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d2a84a149714a8cb3e0c69c87d9a9178d6b8e67623e08abee8c49a1782ff5b4
3
+ size 86709926
vit/sit_vitb_in1k.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:201f38e115ab30ede0f2665a5dace25e94aa66f87a0d631363985d51145a0794
3
+ size 343241740
vit/sit_vits_in1k.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29fcfb461a0fb39121862f309ae680422e478068b1e295589ffc9c4eddc052d8
3
+ size 86709772