File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -396,6 +396,7 @@ def num_parameters(self) -> int:
396396 @abc .abstractmethod
397397 def parameters (self , session_id : Optional [int ] = None ):
398398 """Iterate over all parameters of the model.
399+
399400 Args:
400401 session_id: The session ID, an :py:class:`int` between 0 and
401402 the number of sessions -1 for multisession, and set to
@@ -412,6 +413,7 @@ def _compute_features(
412413 model : Optional [torch .nn .Module ] = None
413414 ) -> Tuple [torch .Tensor , torch .Tensor , torch .Tensor ]:
414415 """Compute the features of the reference, positive and negative samples.
416+
415417 Args:
416418 batch: The input data, not necessarily aligned across the batch
417419 dimension. This means that ``batch.index`` specifies the map
You can’t perform that action at this time.
0 commit comments