modelzoo.common.pytorch.model_utils.checkpoint_converters.helper.tie_none_weights#

modelzoo.common.pytorch.model_utils.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

modelzoo.common.pytorch.model_utils.checkpoint_converters.helper.maybe_tie_lm_head

next

modelzoo.common.pytorch.model_utils.checkpoint_converters.helper.transpose_key_if_2D