From 1e8e028202e3547932e26f6473c6003c4ecded26 Mon Sep 17 00:00:00 2001 From: Xinyu Gong Date: Mon, 6 Apr 2020 22:09:25 -0500 Subject: [PATCH] add a missing 'dilation' args in ReLUConvBN under POOLING --- lib/models/cell_operations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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))