Commit e3759081 authored by Amir Arsalan Soltani's avatar Amir Arsalan Soltani Committed by Soumith Chintala
Browse files

Update resnet.py (#659)

parent fb63374c
......@@ -99,8 +99,8 @@ class Bottleneck(nn.Module):
class ResNet(nn.Module):
def __init__(self, block, layers, num_classes=1000):
self.inplanes = 64
super(ResNet, self).__init__()
self.inplanes = 64
self.conv1 = nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3,
bias=False)
self.bn1 = nn.BatchNorm2d(64)
......
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