We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f9c4fdf commit 610c9d2Copy full SHA for 610c9d2
torchvision/models/segmentation/deeplabv3.py
@@ -80,7 +80,7 @@ def __init__(self, in_channels, atrous_rates, out_channels=256):
80
self.convs = nn.ModuleList(modules)
81
82
self.project = nn.Sequential(
83
- nn.Conv2d(5 * out_channels, out_channels, 1, bias=False),
+ nn.Conv2d(len(self.convs) * out_channels, out_channels, 1, bias=False),
84
nn.BatchNorm2d(out_channels),
85
nn.ReLU(),
86
nn.Dropout(0.5))
0 commit comments