From 3870a1dd63b54e2fd5cd793cd231f3fe7773c5e7 Mon Sep 17 00:00:00 2001 From: "ahassan@azavea.com" Date: Mon, 12 Oct 2020 11:09:18 +0500 Subject: [PATCH] remove hard coded value --- torchvision/models/segmentation/deeplabv3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/segmentation/deeplabv3.py b/torchvision/models/segmentation/deeplabv3.py index ee5c0c7fe64..3624170a235 100644 --- a/torchvision/models/segmentation/deeplabv3.py +++ b/torchvision/models/segmentation/deeplabv3.py @@ -80,7 +80,7 @@ def __init__(self, in_channels, atrous_rates, out_channels=256): self.convs = nn.ModuleList(modules) self.project = nn.Sequential( - nn.Conv2d(5 * out_channels, out_channels, 1, bias=False), + nn.Conv2d(len(self.convs) * out_channels, out_channels, 1, bias=False), nn.BatchNorm2d(out_channels), nn.ReLU(), nn.Dropout(0.5))