Skip to content

Commit 16349a0

Browse files
committed
Fix wrong import in basic_usage.rst
1 parent 1394395 commit 16349a0

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

docs/source/examples/basic_usage.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@ Import several classes from ``torch`` and ``torchjd``:
1919
from torch.optim import SGD
2020
2121
from torchjd import autojac
22-
from torchjd.aggregation import UPGrad, jac_to_grad
22+
from torchjd.aggregation import UPGrad
23+
from torchjd.autojac import jac_to_grad
2324
2425
Define the model and the optimizer, as usual:
2526

0 commit comments

Comments
 (0)