Update deeplab.py

Jessejx · Nov 2, 2018 · ab711a1 · ab711a1
1 parent 1291064
commit ab711a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch/models/deeplab.py b/pytorch/models/deeplab.py
@@ -143,7 +143,7 @@ def __init__(self, block, layers, num_classes=1000, drop_rate=0):
         self.bn1 = nn.BatchNorm2d(64)
         self.relu = nn.ReLU(inplace=True)
         self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)
-        self.layer1 = self._make_layer(block, 64, layers[0]l drop_rate=drop_rate)
+        self.layer1 = self._make_layer(block, 64, layers[0], drop_rate=drop_rate)
         self.layer2 = self._make_layer(block, 128, layers[1], stride=2, drop_rate)
         self.layer3 = self._make_layer(block, 256, layers[2], stride=2, drop_rate)