diff --git a/lib/models/cell_operations.py b/lib/models/cell_operations.py index d8d0077..1a1953e 100644 --- a/lib/models/cell_operations.py +++ b/lib/models/cell_operations.py @@ -114,7 +114,7 @@ class POOLING(nn.Module): if C_in == C_out: self.preprocess = None else: - self.preprocess = ReLUConvBN(C_in, C_out, 1, 1, 0, affine, track_running_stats) + self.preprocess = ReLUConvBN(C_in, C_out, 1, 1, 0, 1, affine, track_running_stats) if mode == 'avg' : self.op = nn.AvgPool2d(3, stride=stride, padding=1, count_include_pad=False) elif mode == 'max': self.op = nn.MaxPool2d(3, stride=stride, padding=1) else : raise ValueError('Invalid mode={:} in POOLING'.format(mode))