def on_before_zero_grad(self, optimizer):
    self.trainer.call_hook('on_before_zero_grad', optimizer)
def on_before_zero_grad(self, optimizer):
    self.trainer.call_hook('on_before_zero_grad', optimizer)