
"""simple docstring"""







import argparse

import torch

from transformers import MobileBertConfig, MobileBertForPreTraining, load_tf_weights_in_mobilebert
from transformers.utils import logging


logging.set_verbosity_info()
def    __lowerCAmelCase    (_UpperCamelCase	,  _UpperCamelCase	,  _UpperCamelCase		):
	# Initialise PyTorch model
	__lowerCAmelCase      :						str						      =				MobileBertConfig.from_json_file(_UpperCamelCase		)
	print(F"Building PyTorch model from configuration: {config}"		)
	__lowerCAmelCase      :						Optional[int]						      =				MobileBertForPreTraining(_UpperCamelCase		)
	# Load weights from tf checkpoint
	__lowerCAmelCase      :						Union[str, Any]						      =				load_tf_weights_in_mobilebert(_UpperCamelCase	,  _UpperCamelCase	,  _UpperCamelCase		)
	# Save pytorch-model
	print(F"Save PyTorch model to {pytorch_dump_path}"		)
	torch.save(model.state_dict()	,  _UpperCamelCase		)


if __name__ == "__main__":
							lowerCamelCase__        =		argparse.ArgumentParser()
							# Required parameters
							parser.add_argument(
							    """--tf_checkpoint_path""", default=None, type=str, required=True, help="""Path to the TensorFlow checkpoint path."""
							)
							parser.add_argument(
							    """--mobilebert_config_file""",
							    default=None,
							    type=str,
							    required=True,
							    help=(
							        """The config json file corresponding to the pre-trained MobileBERT model. \n"""
							        """This specifies the model architecture."""
							    ),
							)
							parser.add_argument(
							    """--pytorch_dump_path""", default=None, type=str, required=True, help="""Path to the output PyTorch model."""
							)
							lowerCamelCase__        =		parser.parse_args()
							convert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.mobilebert_config_file, args.pytorch_dump_path)

"""simple docstring"""







from .glue import glue_convert_examples_to_features, glue_output_modes, glue_processors, glue_tasks_num_labels
from .squad import SquadExample, SquadFeatures, SquadVaProcessor, SquadVaProcessor, squad_convert_examples_to_features
from .utils import DataProcessor, InputExample, InputFeatures, SingleSentenceClassificationProcessor
from .xnli import xnli_output_modes, xnli_processors, xnli_tasks_num_labels