Skip to content

GitLab

Commit 73281b4f authored by vfdev's avatar vfdev Committed by Francisco Massa
Browse files

Update resnet.py (#487) 

nit: replace 4 by expansion
parent 4db0398a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/models/resnet.py
View file @ 73281b4f
...@@ -64,8 +64,8 @@ class Bottleneck(nn.Module): ...@@ -64,8 +64,8 @@ class Bottleneck(nn.Module):
self.conv2 = nn.Conv2d(planes, planes, kernel_size=3, stride=stride, self.conv2 = nn.Conv2d(planes, planes, kernel_size=3, stride=stride,
padding=1, bias=False) padding=1, bias=False)
self.bn2 = nn.BatchNorm2d(planes) self.bn2 = nn.BatchNorm2d(planes)
self.conv3 = nn.Conv2d(planes, planes * 4, kernel_size=1, bias=False) self.conv3 = nn.Conv2d(planes, planes * self.expansion, kernel_size=1, bias=False)
self.bn3 = nn.BatchNorm2d(planes * 4) self.bn3 = nn.BatchNorm2d(planes * self.expansion)
self.relu = nn.ReLU(inplace=True) self.relu = nn.ReLU(inplace=True)
self.downsample = downsample self.downsample = downsample
self.stride = stride self.stride = stride
......
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