TANG 0.0.1

2020-08-30 09:34:23 +08:00
parent 7a85499edf
commit 715a2e64a1
10 changed files with 690 additions and 2 deletions
--- a/model/normalization.py
+++ b/model/normalization.py
@@ -37,7 +37,6 @@ class LayerNorm2d(nn.Module):
    def forward(self, x):
        ln_mean, ln_var = torch.mean(x, dim=[1, 2, 3], keepdim=True), torch.var(x, dim=[1, 2, 3], keepdim=True)
        x = (x - ln_mean) / torch.sqrt(ln_var + self.eps)
-        print(x.size())
        if self.affine:
            return self.channel_gamma * x + self.channel_beta
        return x