Upload hiera_mae.py
Browse files- hiera/hiera_mae.py +2 -2
hiera/hiera_mae.py
CHANGED
@@ -295,7 +295,7 @@ class MaskedAutoencoderHiera(Hiera):
|
|
295 |
# Image Models
|
296 |
|
297 |
@pretrained_model({
|
298 |
-
"mae_in1k": "https://
|
299 |
}, default="mae_in1k")
|
300 |
def mae_hiera_tiny_224(**kwargs):
|
301 |
return MaskedAutoencoderHiera(
|
@@ -304,7 +304,7 @@ def mae_hiera_tiny_224(**kwargs):
|
|
304 |
|
305 |
|
306 |
@pretrained_model({
|
307 |
-
"mae_in1k": "https://
|
308 |
}, default="mae_in1k")
|
309 |
def mae_hiera_small_224(**kwargs):
|
310 |
return MaskedAutoencoderHiera(
|
|
|
295 |
# Image Models
|
296 |
|
297 |
@pretrained_model({
|
298 |
+
"mae_in1k": "https://huggingface.co/merve/hiera-tiny-224-in1k/resolve/main/mae_hiera_tiny_224.pth",
|
299 |
}, default="mae_in1k")
|
300 |
def mae_hiera_tiny_224(**kwargs):
|
301 |
return MaskedAutoencoderHiera(
|
|
|
304 |
|
305 |
|
306 |
@pretrained_model({
|
307 |
+
"mae_in1k": "https://huggingface.co/merve/hiera-small-224-in1k/resolve/main/mae_hiera_small_224.pth",
|
308 |
}, default="mae_in1k")
|
309 |
def mae_hiera_small_224(**kwargs):
|
310 |
return MaskedAutoencoderHiera(
|