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

cerebras.modelzoo.tools.checkpoint_converters.helper.tie_none_weights(old_key, new_key, old_state_dict, new_state_dict, from_index, action_fn_args=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