    def zero_grad(self):
        """
        Zero parameter grads.
        """
        for param_name, param in self.module.named_parameters():
            param.grad = None
Zero parameter grads.
opensora/adaptor/engine.py