def optimizer_zero_grad(self, batch_idx, optimizer, opt_idx):
    self.trainer.accelerator.optimizer_zero_grad(self.trainer.current_epoch, batch_idx, optimizer, opt_idx)
def optimizer_zero_grad(self, batch_idx, optimizer, opt_idx):
    self.trainer.accelerator.optimizer_zero_grad(self.trainer.current_epoch, batch_idx, optimizer, opt_idx)