def initialize(self, stride, bn=False): self.p = M.AvgPool2D(3, stride)
def initialize(self, stride, bn=False): self.bn = bn self.p = M.AvgPool2D(3, stride) if self.bn: self.batch_norm = M.BatchNorm()