in inplace_abn/abn.py [0:0]
def _get_running_stats(self):
running_mean = (
self.running_mean if not self.training or self.track_running_stats else None
)
running_var = (
self.running_var if not self.training or self.track_running_stats else None
)
return running_mean, running_var