Fix docstring

f-dangel · Nov 7, 2023 · 04825b8 · 04825b8
1 parent e4e6b75
commit 04825b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/curvlinops/kfac.py b/curvlinops/kfac.py
@@ -261,7 +261,7 @@ def _compute_kfac(self):
             handle.remove()
 
     def _compute_loss_and_backward(self, output: Tensor, y: Tensor):
-        """Compute the loss and the backward pass(es) required for KFAC.
+        r"""Compute the loss and the backward pass(es) required for KFAC.
 
         Args:
             output: The model's prediction