Update davit.py

3 years ago · 143a239e93
parent 5f0a9e9a9f
commit 143a239e93
1 changed files with 1 additions and 0 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -70,6 +70,7 @@ class ConvPosEnc(nn.Module):
                              to_2tuple(k // 2),
                              groups=dim)
        self.normtype = normtype
        self.norm = nn.Identity()
        if self.normtype == 'batch':
            self.norm = nn.BatchNorm2d(dim)
        elif self.normtype == 'layer':