diff --git a/network/resnet101_asp_oc.py b/network/resnet101_asp_oc.py
index 0e89b28..97abd02 100755
--- 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(