Skip to content

Commit

Permalink
Minor docstring fix
Browse files Browse the repository at this point in the history
  • Loading branch information
fchollet committed Aug 28, 2024
1 parent dcea7f2 commit d4a5116
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions keras/src/legacy/saving/legacy_h5_format.py
Original file line number Diff line number Diff line change
Expand Up @@ -407,8 +407,8 @@ def load_weights_from_hdf5_group(f, model):
def _set_weights(
instance, symbolic_weights, weight_values, name, skip_mismatch=False
):
"""Safely set weights into a model or a layer,
calling load_own_variables if implemented.
"""Safely set weights into a model or a layer.
Args:
instance: Model or layer instance,
symbolic_weights: symbolic tensors representing
Expand All @@ -417,6 +417,7 @@ def _set_weights(
skip_mismatch: Boolean, whether to skip loading of weights
where there is a mismatch in the shape of the weights,
name: name used to identify the group.
Raises:
ValueError: in case of mismatch between provided
model/layer and weights.
Expand Down

0 comments on commit d4a5116

Please sign in to comment.