Andy1621 commited on
Commit
89eb4bc
·
1 Parent(s): 2930b39

Update uniformer.py

Browse files
Files changed (1) hide show
  1. uniformer.py +2 -2
uniformer.py CHANGED
@@ -368,12 +368,12 @@ class Uniformer(nn.Module):
368
  return x
369
 
370
 
371
- def uniformer_s():
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_b():
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)