We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent dea2eef commit adf11f0Copy full SHA for adf11f0
1 file changed
src/modalities/nn/model_initialization/initialization_routines.py
@@ -116,7 +116,7 @@ def get_scaled_initialization(
116
mean (float): Mean of the normal distribution
117
std (float): Standard deviation of the normal distribution used to initialize the other weights
118
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
+ parameter_name_regexes (RegexFilter): List of parameter name regexes to which the initialization
120
should be applied
121
seed (Optional[int]): Random seed for initialization. Defaults to None.
122
0 commit comments