Update resnet101_asp_oc.py

openseg-group · Jun 11, 2019 · 0944790 · 0944790
1 parent 68c0d73
commit 0944790
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/network/resnet101_asp_oc.py b/network/resnet101_asp_oc.py
@@ -66,7 +66,7 @@ def __init__(self, block, layers, num_classes):
         self.context = nn.Sequential(
                 nn.Conv2d(2048, 512, kernel_size=3, stride=1, padding=1),
                 InPlaceABNSync(512),
-                ASP_OC_Module(512, 512)
+                ASP_OC_Module(512, 256)
                 )
         self.cls = nn.Conv2d(512, num_classes, kernel_size=1, stride=1, padding=0, bias=True)
         self.dsn = nn.Sequential(