Skip to content

M
maskrcnn
Commit 0844f577 authored by 任广辉's avatar 任广辉 Committed by Francisco Massa
Browse files
Options

replace 'residual' with 'identity' (#338)

parent f25c6cff
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
maskrcnn_benchmark/modeling/backbone/resnet.py
View file @ 0844f577
...@@ -250,7 +250,7 @@ class BottleneckWithFixedBatchNorm(nn.Module): ...@@ -250,7 +250,7 @@ class BottleneckWithFixedBatchNorm(nn.Module):
self.bn3 = FrozenBatchNorm2d(out_channels) self.bn3 = FrozenBatchNorm2d(out_channels)
def forward(self, x): def forward(self, x):
residual = x identity = x
out = self.conv1(x) out = self.conv1(x)
out = self.bn1(out) out = self.bn1(out)
...@@ -264,9 +264,9 @@ class BottleneckWithFixedBatchNorm(nn.Module): ...@@ -264,9 +264,9 @@ class BottleneckWithFixedBatchNorm(nn.Module):
out = self.bn3(out0) out = self.bn3(out0)
if self.downsample is not None: if self.downsample is not None:
residual = self.downsample(x) identity = self.downsample(x)
out += residual out += identity
out = F.relu_(out) out = F.relu_(out)
return out return out
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment