diff --git a/src/torchjd/autogram/_module_hook_manager.py b/src/torchjd/autogram/_module_hook_manager.py index 082ace69e..4d225a10e 100644 --- a/src/torchjd/autogram/_module_hook_manager.py +++ b/src/torchjd/autogram/_module_hook_manager.py @@ -153,8 +153,6 @@ class AutogramNode(torch.autograd.Function): of the gramian on backward. """ - generate_vmap_rule = True - @staticmethod def forward( gramian_accumulation_phase: BoolRef,