






'''simple docstring'''







import copy
import inspect

from transformers.testing_utils import require_torch, torch_device
from transformers.utils.backbone_utils import BackboneType






@require_torch
class 					a__    :
     '''simple docstring'''







     _snake_case             =  ()
     _snake_case             =  True






     def     _UpperCamelCase   (self    :	Optional[Any]					)  ->       Union[str, Any]:
           __A		   =      self.config_class

           # test default config
           __A		   =      config_class()
           self.assertIsNotNone(lowercase__					)
           __A		   =      ["stem"] + [f"stage{idx}" for idx in range(1				,				len(config.depths					) + 1					)]
           self.assertEqual(config.stage_names				,				lowercase__					)
           self.assertTrue(set(config.out_features					).issubset(set(config.stage_names					)					)					)

           # Test out_features and out_indices are correctly set
           # out_features and out_indices both None
           __A		   =      config_class(out_features=lowercase__				,				out_indices=lowercase__					)
           self.assertEqual(config.out_features				,				[config.stage_names[-1]]					)
           self.assertEqual(config.out_indices				,				[len(config.stage_names					) - 1]					)

           # out_features and out_indices both set
           __A		   =      config_class(out_features=["stem", "stage1"]				,				out_indices=[0, 1]					)
           self.assertEqual(config.out_features				,				["stem", "stage1"]					)
           self.assertEqual(config.out_indices				,				[0, 1]					)

           # Only out_features set
           __A		   =      config_class(out_features=["stage1", "stage3"]					)
           self.assertEqual(config.out_features				,				["stage1", "stage3"]					)
           self.assertEqual(config.out_indices				,				[1, 3]					)

           # Only out_indices set
           __A		   =      config_class(out_indices=[0, 2]					)
           self.assertEqual(config.out_features				,				[config.stage_names[0], config.stage_names[2]]					)
           self.assertEqual(config.out_indices				,				[0, 2]					)

           # Error raised when out_indices do not correspond to out_features
           with self.assertRaises(lowercase__					):
                 __A		   =      config_class(out_features=["stage1", "stage2"]				,				out_indices=[0, 2]					)






     def     _UpperCamelCase   (self    :	Optional[int]					)  ->       Optional[Any]:
           __A   ,				__A		   =      self.model_tester.prepare_config_and_inputs_for_common()

           for model_class in self.all_model_classes:
                 __A		   =      model_class(lowercase__					)
                 __A		   =      inspect.signature(model.forward					)
                 # signature.parameters is an OrderedDict => so arg_names order is deterministic
                 __A		   =      [*signature.parameters.keys()]
                 __A		   =      ["pixel_values"]
                 self.assertListEqual(arg_names[:1]				,				lowercase__					)






     def     _UpperCamelCase   (self    :	Dict					)  ->       Union[str, Any]:
           __A   ,				__A		   =      self.model_tester.prepare_config_and_inputs_for_common()

           for model_class in self.all_model_classes:
                 __A		   =      model_class(lowercase__					)
                 self.assertEqual(len(model.channels					)				,				len(config.out_features					)					)
                 __A		   =      model.num_features
                 __A		   =      [config.stage_names.index(lowercase__					) for feat in config.out_features]
                 __A		   =      [num_features[idx] for idx in out_indices]
                 self.assertListEqual(model.channels				,				lowercase__					)

                 __A		   =      copy.deepcopy(lowercase__					)
                 __A		   =      None
                 __A		   =      model_class(lowercase__					)
                 self.assertEqual(len(model.channels					)				,				1					)
                 self.assertListEqual(model.channels				,				[num_features[-1]]					)

                 __A		   =      copy.deepcopy(lowercase__					)
                 __A		   =      None
                 __A		   =      model_class(lowercase__					)
                 self.assertEqual(len(model.channels					)				,				1					)
                 self.assertListEqual(model.channels				,				[num_features[-1]]					)






     def     _UpperCamelCase   (self    :	Optional[Any]					)  ->       Optional[Any]:
           __A   ,				__A		   =      self.model_tester.prepare_config_and_inputs_for_common()

           for model_class in self.all_model_classes:
                 __A		   =      model_class(lowercase__					)
                 model.to(lowercase__					)
                 model.eval()
                 __A		   =      model(**lowercase__					)

                 self.assertEqual(len(result.feature_maps					)				,				len(config.out_features					)					)
                 self.assertEqual(len(model.channels					)				,				len(config.out_features					)					)
                 self.assertEqual(len(result.feature_maps					)				,				len(config.out_indices					)					)
                 self.assertEqual(len(model.channels					)				,				len(config.out_indices					)					)

                 # Check output of last stage is taken if out_features=None, out_indices=None
                 __A		   =      copy.deepcopy(lowercase__					)
                 __A		   =      None
                 __A		   =      model_class(lowercase__					)
                 model.to(lowercase__					)
                 model.eval()
                 __A		   =      model(**lowercase__					)

                 self.assertEqual(len(result.feature_maps					)				,				1					)
                 self.assertEqual(len(model.channels					)				,				1					)

                 __A		   =      copy.deepcopy(lowercase__					)
                 __A		   =      None
                 __A		   =      model_class(lowercase__					)
                 model.to(lowercase__					)
                 model.eval()
                 __A		   =      model(**lowercase__					)

                 self.assertEqual(len(result.feature_maps					)				,				1					)
                 self.assertEqual(len(model.channels					)				,				1					)

                 # Check backbone can be initialized with fresh weights
                 __A		   =      copy.deepcopy(lowercase__					)
                 __A		   =      False
                 __A		   =      model_class(lowercase__					)
                 model.to(lowercase__					)
                 model.eval()
                 __A		   =      model(**lowercase__					)






     def     _UpperCamelCase   (self    :	Dict					)  ->       Optional[int]:
           __A   ,				__A		   =      self.model_tester.prepare_config_and_inputs_for_common()

           for backbone_class in self.all_model_classes:
                 __A		   =      backbone_class(lowercase__					)

                 self.assertTrue(hasattr(lowercase__				,				"backbone_type"					)					)
                 self.assertTrue(hasattr(lowercase__				,				"stage_names"					)					)
                 self.assertTrue(hasattr(lowercase__				,				"num_features"					)					)
                 self.assertTrue(hasattr(lowercase__				,				"out_indices"					)					)
                 self.assertTrue(hasattr(lowercase__				,				"out_features"					)					)
                 self.assertTrue(hasattr(lowercase__				,				"out_feature_channels"					)					)
                 self.assertTrue(hasattr(lowercase__				,				"channels"					)					)

                 self.assertIsInstance(backbone.backbone_type				,				lowercase__					)
                 # Verify num_features has been initialized in the backbone init
                 self.assertIsNotNone(backbone.num_features					)
                 self.assertTrue(len(backbone.channels					) == len(backbone.out_indices					)					)
                 self.assertTrue(len(backbone.stage_names					) == len(backbone.num_features					)					)
                 self.assertTrue(len(backbone.channels					) <= len(backbone.num_features					)					)
                 self.assertTrue(len(backbone.out_feature_channels					) == len(backbone.stage_names					)					)






     def     _UpperCamelCase   (self    :	Tuple					)  ->       int:
           __A   ,				__A		   =      self.model_tester.prepare_config_and_inputs_for_common()
           __A		   =      inputs_dict["pixel_values"].shape[0]

           for backbone_class in self.all_model_classes:
                 __A		   =      backbone_class(lowercase__					)
                 backbone.to(lowercase__					)
                 backbone.eval()

                 __A		   =      backbone(**lowercase__					)

                 # Test default outputs and verify feature maps
                 self.assertIsInstance(outputs.feature_maps				,				lowercase__					)
                 self.assertTrue(len(outputs.feature_maps					) == len(backbone.channels					)					)
                 for feature_map, n_channels in zip(outputs.feature_maps				,				backbone.channels					):
                       self.assertTrue(feature_map.shape[:2]				,				(batch_size, n_channels)					)
                 self.assertIsNone(outputs.hidden_states					)
                 self.assertIsNone(outputs.attentions					)

                 # Test output_hidden_states=True
                 __A		   =      backbone(**lowercase__				,				output_hidden_states=lowercase__					)
                 self.assertIsNotNone(outputs.hidden_states					)
                 self.assertTrue(len(outputs.hidden_states					)				,				len(backbone.stage_names					)					)
                 for hidden_state, n_channels in zip(outputs.hidden_states				,				backbone.channels					):
                       self.assertTrue(hidden_state.shape[:2]				,				(batch_size, n_channels)					)

                 # Test output_attentions=True
                 if self.has_attentions:
                       __A		   =      backbone(**lowercase__				,				output_attentions=lowercase__					)
                       self.assertIsNotNone(outputs.attentions					)









'''simple docstring'''







import importlib
import sys
from argparse import REMAINDER, ArgumentParser
from pathlib import Path

import torch_xla.distributed.xla_multiprocessing as xmp


def 						_UpperCamelCase							(     ):
      __A		   =      ArgumentParser(
          description=(
              "PyTorch TPU distributed training launch helper utility that will spawn up multiple distributed processes"
          )							)

      # Optional arguments for the launch helper
      parser.add_argument("--num_cores"							,			type=snake_case__							,			default=1							,			help="Number of TPU cores to use (1 or 8)."							)

      # positional
      parser.add_argument(
          "training_script"							,			type=snake_case__							,			help=(
              "The full path to the single TPU training "
              "program/script to be launched in parallel, "
              "followed by all the arguments for the "
              "training script"
          )							,			)

      # rest from the training program
      parser.add_argument("training_script_args"							,			nargs=snake_case__							)

      return parser.parse_args()


def 						_UpperCamelCase							(     ):
      __A		   =      parse_args()

      # Import training_script as a module.
      __A		   =      Path(args.training_script							)
      sys.path.append(str(script_fpath.parent.resolve()							)							)
      __A		   =      script_fpath.stem
      __A		   =      importlib.import_module(snake_case__							)

      # Patch sys.argv
      __A		   =      [args.training_script] + args.training_script_args + ["--tpu_num_cores", str(args.num_cores							)]

      xmp.spawn(mod._mp_fn							,			args=()							,			nprocs=args.num_cores							)


if __name__ == "__main__":
       main()


