From da431dd68f4584a02ce1fe93947aa5fc5486e654 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 25 Apr 2021 22:06:10 +0200 Subject: [PATCH] Update activations.py --- utils/activations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/activations.py b/utils/activations.py index 3863a1b03051..75d286afb8f6 100644 --- a/utils/activations.py +++ b/utils/activations.py @@ -86,8 +86,8 @@ def __init__(self, c1, k=1, s=1, r=16): # ch_in, kernel, stride, r self.p2 = nn.Parameter(torch.randn(1, c1, 1, 1)) self.fc1 = nn.Conv2d(c1, c2, k, s, bias=False) self.fc2 = nn.Conv2d(c2, c1, k, s, bias=False) - # self.bn2 = nn.BatchNorm2d(c1) # self.bn1 = nn.BatchNorm2d(c2) + # self.bn2 = nn.BatchNorm2d(c1) def forward(self, x): y = x.mean(dim=2, keepdims=True).mean(dim=3, keepdims=True)