We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 96ece7a commit 8ee2ed4Copy full SHA for 8ee2ed4
tests/utils/forward_backwards.py
@@ -6,7 +6,7 @@
6
from torch.utils._pytree import PyTree, tree_flatten, tree_map
7
8
from torchjd.aggregation import Aggregator, Weighting
9
-from torchjd.autogram._engine import Engine
+from torchjd.autogram import Engine
10
from torchjd.autojac import backward
11
12
0 commit comments