def METHOD_NAME(
        self,
        state_dict: dict,
        prefix: str,
        local_metadata: dict,
        strict: bool,
        missing_keys: List[str],
        unexpected_keys: List[str],
        error_msgs: List[str], ):
    num_batches_tracked_key = prefix + "num_batches_tracked"
    if num_batches_tracked_key in state_dict:
        del state_dict[num_batches_tracked_key]
    super().METHOD_NAME(state_dict, prefix, local_metadata,
                                  strict, missing_keys, unexpected_keys,
                                  error_msgs)