Skip to content

Commit 610c9d2

Browse files
AdeelHvfdev-5
andauthored
remove hard coded value (#2793)
Co-authored-by: vfdev <[email protected]>
1 parent f9c4fdf commit 610c9d2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/models/segmentation/deeplabv3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def __init__(self, in_channels, atrous_rates, out_channels=256):
8080
self.convs = nn.ModuleList(modules)
8181

8282
self.project = nn.Sequential(
83-
nn.Conv2d(5 * out_channels, out_channels, 1, bias=False),
83+
nn.Conv2d(len(self.convs) * out_channels, out_channels, 1, bias=False),
8484
nn.BatchNorm2d(out_channels),
8585
nn.ReLU(),
8686
nn.Dropout(0.5))

0 commit comments

Comments
 (0)