Skip to content

Commit b6c6d8b

Browse files
committed
Fix
1 parent df450a7 commit b6c6d8b

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/torchjd/aggregation/_gradvac.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,10 @@ class GradVac(GramianWeightedAggregator, Stochastic):
161161
:param eps: Small non-negative constant added to denominators.
162162
:param seed: Seed for the internal random number generator. If ``None``, a seed is drawn from
163163
the global PyTorch RNG to fork an independent stream.
164+
165+
.. note::
166+
To apply GradVac with the `whole_model`, `enc_dec`, `all_layer` or `all_matrix` grouping
167+
strategy, please refer to the :doc:`Grouping </examples/grouping>` examples.
164168
"""
165169

166170
def __init__(self, beta: float = 0.5, eps: float = 1e-8, seed: int | None = None) -> None:

0 commit comments

Comments
 (0)