Spaces:
Runtime error
Runtime error
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)
|