Spaces:
Sleeping
Sleeping
Update uniformer.py
Browse files- uniformer.py +2 -2
uniformer.py
CHANGED
@@ -368,12 +368,12 @@ class Uniformer(nn.Module):
|
|
368 |
return x
|
369 |
|
370 |
|
371 |
-
def
|
372 |
return Uniformer(
|
373 |
depth=[3, 4, 8, 3], embed_dim=[64, 128, 320, 512],
|
374 |
head_dim=64, drop_rate=0.1)
|
375 |
|
376 |
-
def
|
377 |
return Uniformer(
|
378 |
depth=[5, 8, 20, 7], embed_dim=[64, 128, 320, 512],
|
379 |
head_dim=64, drop_rate=0.3)
|
|
|
368 |
return x
|
369 |
|
370 |
|
371 |
+
def uniformer_small():
|
372 |
return Uniformer(
|
373 |
depth=[3, 4, 8, 3], embed_dim=[64, 128, 320, 512],
|
374 |
head_dim=64, drop_rate=0.1)
|
375 |
|
376 |
+
def uniformer_base():
|
377 |
return Uniformer(
|
378 |
depth=[5, 8, 20, 7], embed_dim=[64, 128, 320, 512],
|
379 |
head_dim=64, drop_rate=0.3)
|