cerebras.modelzoo.tools.checkpoint_converters.helper.transpose_key_if_2D#

cerebras.modelzoo.tools.checkpoint_converters.helper.transpose_key_if_2D(old_key, new_key, old_state_dict, new_state_dict, from_index, action_fn_args)[source]#