def METHOD_NAME(self) -> None:
    # check grads are scaled
    scale = self.trainer.precision_plugin.scaler.get_scale()
    assert scale != 1.0  # the return value if not enabled
    grads = [p.grad for p in self.parameters()]
    inv_scale = 1 / scale
    self.original_grads = [p * inv_scale for p in grads]