Skip to content

Commit

Permalink
remove hard coded value (pytorch#2793)
Browse files Browse the repository at this point in the history
Co-authored-by: vfdev <vfdev.5@gmail.com>
  • Loading branch information
AdeelH and vfdev-5 committed Dec 4, 2020
1 parent 9d84eb8 commit cc702fe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchvision/models/segmentation/deeplabv3.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand Down

0 comments on commit cc702fe

Please sign in to comment.