Ejemplo n.º 1
0
 def forward(self, x):
     output = inplace_abn(x, self.weight, self.bias, self.running_mean,
                          self.running_var, self.training, self.momentum,
                          self.eps, self.act_name, self.act_param)
     if isinstance(output, tuple):
         output = output[0]
     return output
Ejemplo n.º 2
0
 def inplace_abn_sync(**kwargs):
     inplace_abn(**kwargs)