Skip to content

Commit adf11f0

Browse files
rrutmannle1nux
andauthored
fix: Use correct type for parameter_name_regexes
Co-authored-by: Max Lübbering <2804731+le1nux@users.noreply.github.com>
1 parent dea2eef commit adf11f0

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/modalities/nn/model_initialization/initialization_routines.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def get_scaled_initialization(
116116
mean (float): Mean of the normal distribution
117117
std (float): Standard deviation of the normal distribution used to initialize the other weights
118118
num_layers (int): Number of layers in the model which we use to downscale std with
119-
parameter_name_regexes (list[str]): List of parameter name regexes to which the initialization
119+
parameter_name_regexes (RegexFilter): List of parameter name regexes to which the initialization
120120
should be applied
121121
seed (Optional[int]): Random seed for initialization. Defaults to None.
122122

0 commit comments

Comments
 (0)