Skip to content

Commit

Permalink
Precommit fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
sanaAyrml committed Sep 24, 2024
1 parent fdc5c9e commit 5758329
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fl4health/model_bases/feature_extractor_buffer.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def find_last_common_prefix(self, prefix: str, layers_name: List[str]) -> str:
Args:
prefix (str): The prefix of the layer name for registering the hook.
layers_name (List[str]): The list of named modules of the model. The assumption is that list of
layers_name (List[str]): The list of named modules of the model. The assumption is that list of
named modules is sorted in the order of the model's forward pass with depth-first traversal. This
will allow the user to specify the generic name of the layer instead of the full hierarchical name.
Expand Down

0 comments on commit 5758329

Please sign in to comment.