From ffe384255c29217deb5c3075d56f77f649d339fd Mon Sep 17 00:00:00 2001 From: Zhanghan Ke Date: Fri, 8 Jan 2021 17:50:01 +0800 Subject: [PATCH] bug fix --- src/models/modnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/models/modnet.py b/src/models/modnet.py index 16609b3..9e268e7 100644 --- a/src/models/modnet.py +++ b/src/models/modnet.py @@ -24,7 +24,7 @@ class IBNorm(nn.Module): def forward(self, x): bn_x = self.bnorm(x[:, :self.bnorm_channels, ...].contiguous()) - in_x = self.inorm(x[:, self.inorm_channels:, ...].contiguous()) + in_x = self.inorm(x[:, self.bnorm_channels:, ...].contiguous()) return torch.cat((bn_x, in_x), 1)