We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
GradDrop
1 parent 40842c4 commit 66ca700Copy full SHA for 66ca700
src/torchjd/aggregation/graddrop.py
@@ -72,8 +72,8 @@ def _check_matrix_has_enough_rows(self, matrix: Tensor) -> None:
72
n_rows = matrix.shape[0]
73
if self.leak is not None and n_rows != len(self.leak):
74
raise ValueError(
75
- f"Parameter `matrix` should be a matrix of at least {len(self.leak)} rows "
76
- f"(i.e. the number of leak scalars). Found `matrix` of shape `{matrix.shape}`."
+ f"Parameter `matrix` should be a matrix of exactly {len(self.leak)} rows (i.e. the "
+ f"number of leak scalars). Found `matrix` of shape `{matrix.shape}`."
77
)
78
79
def __repr__(self) -> str:
0 commit comments