mirror
/
MODNet
mirror of https://github.com/ZHKKKe/MODNet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug fix

pull/78/head
Zhanghan Ke 2021-01-08 17:50:01 +08:00 committed by GitHub
parent bece9b41af
commit ffe384255c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/modnet.py
View File

@ -24,7 +24,7 @@ class IBNorm(nn.Module):
def forward(self, x): def forward(self, x):
bn_x = self.bnorm(x[:, :self.bnorm_channels, ...].contiguous()) 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) return torch.cat((bn_x, in_x), 1)