cerebras.modelzoo.tools.checkpoint_converters.helper.tie_none_weights#

cerebras.modelzoo.tools.checkpoint_converters.helper.tie_none_weights(old_key: str, new_key: str, old_state_dict: collections.OrderedDict, new_state_dict: collections.OrderedDict, from_index: int, action_fn_args: Optional[dict] = None) → None[source]#

Ties weights stored at old_key & new_key

previous

cerebras.modelzoo.tools.checkpoint_converters.helper.maybe_tie_lm_head

next

cerebras.modelzoo.tools.checkpoint_converters.helper.transpose_key_if_2D