





def      _lowerCAmelCase		(	_lowerCAmelCase  ):
   '''simple docstring'''


   if not isinstance(a__	,a__  ) or number < 0:
      raise ValueError("""Input must be a non-negative integer"""  )

   A_  :					Optional[Any]         =						0
   while number:
      # This way we arrive at next set bit (next 1) instead of looping
      # through each bit and checking for 1s hence the
      # loop won't run 32 times it will only run the number of `1` times
      number &= number - 1
      count += 1
   return count


if __name__ == "__main__":
      import doctest

      doctest.testmod()














"""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        UpperCAmelCase      (					a__     ,      a__     ,      a__ ):





						'''simple docstring'''






						lowerCAmelCase   :Tuple						=			MobileBertConfig.from_json_file(a__ )
						print(F"""Building PyTorch model from configuration: {config}""" )
						lowerCAmelCase   :Tuple						=			MobileBertForPreTraining(a__ )
						# Load weights from tf checkpoint
						lowerCAmelCase   :Any						=			load_tf_weights_in_mobilebert(a__     ,      a__     ,      a__ )
						# Save pytorch-model
						print(F"""Save PyTorch model to {pytorch_dump_path}""" )
						torch.save(model.state_dict()     ,      a__ )


if __name__ == "__main__":
		__SCREAMING_SNAKE_CASE   =   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.'
		)
		__SCREAMING_SNAKE_CASE   =   parser.parse_args()
		convert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.mobilebert_config_file, args.pytorch_dump_path)