Skip to content

Commit

Permalink
[FIX] Long lines
Browse files Browse the repository at this point in the history
  • Loading branch information
f-dangel committed Mar 2, 2024
1 parent 23d408f commit 57c30a8
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions docs_src/examples/use_cases/example_custom_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@ def forward(self, input: torch.Tensor) -> torch.Tensor:
#
# First-order extensions
# ----------------------
# First we focus on `BackPACK's first-order extensions <https://docs.backpack.pt/en/master/extensions.html#first-order-extensions>`_.
# First we focus on `BackPACK's first-order extensions
# <https://docs.backpack.pt/en/master/extensions.html#first-order-extensions>`_.
# They don't backpropagate additional information and thus require less functionality.
#
# Let's make BackPACK support computing individual gradients for ``ScaleModule``.
Expand Down Expand Up @@ -239,9 +240,11 @@ def weight(
# %%
# Second-order extension
# ----------------------
# Next, we focus on `BackPACK's second-order extensions <https://docs.backpack.pt/en/master/extensions.html#second-order-extensions>`_.
# They backpropagate additional information and thus require more functionality to be implemented and a more in-depth
# understanding of BackPACK's internals and the quantity of interest.
# Next, we focus on `BackPACK's second-order extensions
# <https://docs.backpack.pt/en/master/extensions.html#second-order-extensions>`_.
# They backpropagate additional information and thus require more functionality to be
# implemented and a more in-depth understanding of BackPACK's internals and
# the quantity of interest.
#
# Let's make BackPACK support computing the exact diagonal of the generalized
# Gauss-Newton (GGN) matrix
Expand Down

0 comments on commit 57c30a8

Please sign in to comment.