Skip to content

GitLab

Unverified Commit 6cd90e6e authored by eellison's avatar eellison Committed by GitHub
Browse files

fix resnext (#1575)

parent 44a5bae9
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
torchvision/models/segmentation/deeplabv3.py
View file @ 6cd90e6e
...@@ -57,7 +57,8 @@ class ASPPPooling(nn.Sequential): ...@@ -57,7 +57,8 @@ class ASPPPooling(nn.Sequential):
def forward(self, x): def forward(self, x):
size = x.shape[-2:] size = x.shape[-2:]
x = super(ASPPPooling, self).forward(x) for mod in self:
x = mod(x)
return F.interpolate(x, size=size, mode='bilinear', align_corners=False) return F.interpolate(x, size=size, mode='bilinear', align_corners=False)
......
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