From f88078d20b55648a2752290541305c1a3eb4a22f Mon Sep 17 00:00:00 2001 From: Bobby V Date: Sat, 29 May 2021 14:19:09 -0400 Subject: [PATCH] Update PSPnet. typo in the code The net can't compile due to a typo. --- pytorch/models/pspnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch/models/pspnet.py b/pytorch/models/pspnet.py index 57f45d9..83f7f60 100644 --- a/pytorch/models/pspnet.py +++ b/pytorch/models/pspnet.py @@ -62,7 +62,7 @@ class Bottleneck(nn.Module): def __init__(self, inplanes, planes, stride=1, downsample=None, dilation=1, drop_rate=0): super(Bottleneck, self).__init__() self.dp = nn.Dropout2d(p=drop_rate) - elf.conv1 = nn.Conv2d(inplanes, planes, kernel_size=1, bias=False) + self.conv1 = nn.Conv2d(inplanes, planes, kernel_size=1, bias=False) self.bn1 = nn.BatchNorm2d(planes) self.conv2 = nn.Conv2d(planes, planes, kernel_size=3, stride=stride, dilation=dilation, padding=dilation, bias=False)