Ejemplo n.º 1
0
 def initialize(self, stride, bn=False):
     self.p = M.AvgPool2D(3, stride)
Ejemplo n.º 2
0
	def initialize(self, stride, bn=False):
		self.bn = bn 
		self.p = M.AvgPool2D(3, stride)
		if self.bn:
			self.batch_norm = M.BatchNorm()