Skip to content

Commit

Permalink
[FIX] RTD examples
Browse files Browse the repository at this point in the history
  • Loading branch information
f-dangel committed Nov 4, 2024
1 parent d63a896 commit fc67b7f
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
4 changes: 3 additions & 1 deletion docs/examples/basic_usage/example_eigenvalues.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,9 @@ def orthonormalize(v: numpy.ndarray, basis: List[numpy.ndarray]) -> numpy.ndarra
# the linear operator's progress bar, which allows us to count the number of
# matrix-vector products invoked by both eigen-solvers:

H = HessianLinearOperator(model, loss_function, params, data, progressbar=True)
H = HessianLinearOperator(
model, loss_function, params, data, progressbar=True
).to_scipy()

# determine number of matrix-vector products used by `eigsh`
with StringIO() as buf, redirect_stderr(buf):
Expand Down
2 changes: 1 addition & 1 deletion docs/examples/basic_usage/example_visual_tour.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@
params,
dataloader,
block_sizes=[s for s in num_tensors_layer if s != 0],
)
).to_scipy()
F_linop = FisherMCLinearOperator(model, loss_function, params, dataloader)
KFAC_linop = KFACLinearOperator(
model, loss_function, params, dataloader, separate_weight_and_bias=False
Expand Down

0 comments on commit fc67b7f

Please sign in to comment.