modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18#
- class modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18[source]#
-
Methods
convert
convert_all_keys
Converts all keys in a checkpoint from from_index format to the other format.
Attempts to convert the old key by matching against the list of conversion rules.
converter_note
Hook to extract model state dicts out of the input/output checkpoint
file_formats
formats
get_config_converter_class
get_from_index
(Pre)Initializes the output checkpoint at a supplied path.
load
Hook executes after checkpoint conversion.
Hook executes right after model conversion.
Hook executes before checkpoint conversion.
Hook executes right before model conversion.
Copies value that exists at old_state_dict's old_key to new_state_dict's new_key.
save
supports_conversion
- convert_helper(input_checkpoint, configs: Tuple[dict, dict], from_index: int, output_checkpoint={}, drop_unmatched_keys: bool = False, no_progress_bar: bool = True, debug: bool = False)#
Converts all keys in a checkpoint from from_index format to the other format. Conversion will fail if at least one of the keys did not match on any conversion rules and drop_unmatched_keys is not enabled. Returns the newly converted checkpoint.
- convert_key(old_key: str, old_state_dict: collections.OrderedDict, new_state_dict: collections.OrderedDict, from_index: int, match_start: int = 0, prefix: str = '', action_fn_args: Optional[dict] = None, debug: bool = False) None #
Attempts to convert the old key by matching against the list of conversion rules. The first rule to match is used for conversion (i.e. even if multiple rules would match, the latter ones are never used). Returns True if a conversion occurred.
- extract_model_dict(input_checkpoint, output_checkpoint, configs: Tuple[dict, dict], from_index: int)#
Hook to extract model state dicts out of the input/output checkpoint
- classmethod init_output_checkpoint(file_without_ext: str, from_index: int, hf_shard_size: Union[str, int] = '10GB', **kwargs) str #
(Pre)Initializes the output checkpoint at a supplied path. This is used in streaming conversion when the checkpoint is written to file as conversion is performed rather than accumulating the full checkpoint in memory and saving to file at the very end.
- post_checkpoint_convert(input_checkpoint, output_checkpoint, configs: Tuple[dict, dict], from_index: int)#
Hook executes after checkpoint conversion.
- post_model_convert(old_state_dict: collections.OrderedDict, new_state_dict: collections.OrderedDict, configs: Tuple[dict, dict], from_index: int, drop_unmatched_keys: bool)#
Hook executes right after model conversion.
- pre_checkpoint_convert(input_checkpoint, output_checkpoint, configs: Tuple[dict, dict], from_index: int)#
Hook executes before checkpoint conversion.
- pre_model_convert(old_state_dict: collections.OrderedDict, new_state_dict: collections.OrderedDict, configs: Tuple[dict, dict], from_index: int, drop_unmatched_keys: bool)#
Hook executes right before model conversion.
- static replaceKey(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 #
Copies value that exists at old_state_dict’s old_key to new_state_dict’s new_key.