


'''simple docstring'''




from collections import OrderedDict
from typing import List, Mapping

from packaging import version

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


UpperCAmelCase_      :		Tuple          =		logging.get_logger(__name__)

UpperCAmelCase_      :		Dict          =		{
    '''google/efficientnet-b7''': '''https://huggingface.co/google/efficientnet-b7/resolve/main/config.json''',
}







class 					a			(				__a  ):
 '''simple docstring'''




 __lowerCAmelCase						: Union[str, Any]							  =		'efficientnet'




 def __init__(    self		,      lowerCamelCase_ = 3		,      lowerCamelCase_ = 6_0_0		,      lowerCamelCase_ = 2.0		,      lowerCamelCase_ = 3.1		,      lowerCamelCase_ = 8		,      lowerCamelCase_ = [3, 3, 5, 3, 5, 5, 3]		,      lowerCamelCase_ = [3_2, 1_6, 2_4, 4_0, 8_0, 1_1_2, 1_9_2]		,      lowerCamelCase_ = [1_6, 2_4, 4_0, 8_0, 1_1_2, 1_9_2, 3_2_0]		,      lowerCamelCase_ = []		,      lowerCamelCase_ = [1, 2, 2, 2, 1, 2, 1]		,      lowerCamelCase_ = [1, 2, 2, 3, 3, 4, 1]		,      lowerCamelCase_ = [1, 6, 6, 6, 6, 6, 6]		,      lowerCamelCase_ = 0.25		,      lowerCamelCase_ = "swish"		,      lowerCamelCase_ = 2_5_6_0		,      lowerCamelCase_ = "mean"		,      lowerCamelCase_ = 0.02		,      lowerCamelCase_ = 0.001		,      lowerCamelCase_ = 0.99		,      lowerCamelCase_ = 0.5		,      lowerCamelCase_ = 0.2		,      **lowerCamelCase_		,      )      ->    Optional[int]:
    super().__init__(**_lowerCamelCase      )

    _a					:  List[Any]						=  num_channels
    _a					:  Dict						=  image_size
    _a					:  Union[str, Any]						=  width_coefficient
    _a					:  List[Any]						=  depth_coefficient
    _a					:  Dict						=  depth_divisor
    _a					:  List[Any]						=  kernel_sizes
    _a					:  Optional[Any]						=  in_channels
    _a					:  Tuple						=  out_channels
    _a					:  int						=  depthwise_padding
    _a					:  List[Any]						=  strides
    _a					:  Optional[int]						=  num_block_repeats
    _a					:  str						=  expand_ratios
    _a					:  Optional[Any]						=  squeeze_expansion_ratio
    _a					:  Union[str, Any]						=  hidden_act
    _a					:  List[str]						=  hidden_dim
    _a					:  Tuple						=  pooling_type
    _a					:  Union[str, Any]						=  initializer_range
    _a					:  Dict						=  batch_norm_eps
    _a					:  List[Any]						=  batch_norm_momentum
    _a					:  List[str]						=  dropout_rate
    _a					:  Tuple						=  drop_connect_rate
    _a					:  Optional[Any]						=  sum(_lowerCamelCase      ) * 4







class 					a			(				__a  ):
 '''simple docstring'''




 __lowerCAmelCase						: str							  =		version.parse("""1.11"""  )




 @property
 def   __UpperCamelCase   (    self      )      ->    Tuple:
    return OrderedDict(
        [
            ('pixel_values', {0: 'batch', 1: 'num_channels', 2: 'height', 3: 'width'}),
        ]      )




 @property
 def   __UpperCamelCase   (    self      )      ->    Tuple:
    return 1e-5



'''simple docstring'''

import os
import tempfile
import unittest
from pathlib import Path

from transformers import AutoConfig, is_torch_available
from transformers.testing_utils import require_torch, torch_device


if is_torch_available():
 from transformers import PyTorchBenchmark, PyTorchBenchmarkArguments


@require_torch
class     __UpperCAmelCase       ( unittest.TestCase		):





  def 						UpperCAmelCase_	(	self    ,     _lowerCamelCase ):
     for model_result in results.values():
        for batch_size, sequence_length in zip(model_result['''bs''']    ,     model_result['''ss'''] ):
           lowerCAmelCase_			=      model_result['''result'''][batch_size][sequence_length]
           self.assertIsNotNone(_lowerCamelCase )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_inference_result )
     self.check_results_dict_not_empty(results.memory_inference_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sgugger/tiny-distilbert-classification'''
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     only_pretrain_model=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_inference_result )
     self.check_results_dict_not_empty(results.memory_inference_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     torchscript=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_inference_result )
     self.check_results_dict_not_empty(results.memory_inference_result )





  @unittest.skipIf(torch_device == '''cpu'''    ,     '''Cant do half precision''' )
  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     fpaa=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_inference_result )
     self.check_results_dict_not_empty(results.memory_inference_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      AutoConfig.from_pretrained(_lowerCamelCase )
     # set architectures equal to `None`
     lowerCAmelCase_			=      None
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase    ,     configs=[config] )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_inference_result )
     self.check_results_dict_not_empty(results.memory_inference_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_train_result )
     self.check_results_dict_not_empty(results.memory_train_result )





  @unittest.skipIf(torch_device == '''cpu'''    ,     '''Can\'t do half precision''' )
  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     fpaa=_lowerCamelCase    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_train_result )
     self.check_results_dict_not_empty(results.memory_train_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      AutoConfig.from_pretrained(_lowerCamelCase )
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase    ,     configs=[config] )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_inference_result )
     self.check_results_dict_not_empty(results.memory_inference_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tinier_bart'''
     lowerCAmelCase_			=      AutoConfig.from_pretrained(_lowerCamelCase )
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase    ,     configs=[config] )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_inference_result )
     self.check_results_dict_not_empty(results.memory_inference_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     lowerCAmelCase_			=      AutoConfig.from_pretrained(_lowerCamelCase )
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase    ,     configs=[config] )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_train_result )
     self.check_results_dict_not_empty(results.memory_train_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tinier_bart'''
     lowerCAmelCase_			=      AutoConfig.from_pretrained(_lowerCamelCase )
     lowerCAmelCase_			=      PyTorchBenchmarkArguments(
         models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     multi_process=_lowerCamelCase    ,     )
     lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase    ,     configs=[config] )
     lowerCAmelCase_			=      benchmark.run()
     self.check_results_dict_not_empty(results.time_train_result )
     self.check_results_dict_not_empty(results.memory_train_result )





  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''
     with tempfile.TemporaryDirectory() as tmp_dir:
        lowerCAmelCase_			=      PyTorchBenchmarkArguments(
            models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     save_to_csv=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     inference_time_csv_file=os.path.join(_lowerCamelCase    ,     '''inf_time.csv''' )    ,     train_memory_csv_file=os.path.join(_lowerCamelCase    ,     '''train_mem.csv''' )    ,     inference_memory_csv_file=os.path.join(_lowerCamelCase    ,     '''inf_mem.csv''' )    ,     train_time_csv_file=os.path.join(_lowerCamelCase    ,     '''train_time.csv''' )    ,     env_info_csv_file=os.path.join(_lowerCamelCase    ,     '''env.csv''' )    ,     multi_process=_lowerCamelCase    ,     )
        lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
        benchmark.run()
        self.assertTrue(Path(os.path.join(_lowerCamelCase    ,     '''inf_time.csv''' ) ).exists() )
        self.assertTrue(Path(os.path.join(_lowerCamelCase    ,     '''train_time.csv''' ) ).exists() )
        self.assertTrue(Path(os.path.join(_lowerCamelCase    ,     '''inf_mem.csv''' ) ).exists() )
        self.assertTrue(Path(os.path.join(_lowerCamelCase    ,     '''train_mem.csv''' ) ).exists() )
        self.assertTrue(Path(os.path.join(_lowerCamelCase    ,     '''env.csv''' ) ).exists() )







  def 						UpperCAmelCase_	(	self ):
     lowerCAmelCase_			=      '''sshleifer/tiny-gpt2'''

     def _check_summary_is_not_empty(_lowerCamelCase ):
        self.assertTrue(hasattr(_lowerCamelCase    ,     '''sequential''' ) )
        self.assertTrue(hasattr(_lowerCamelCase    ,     '''cumulative''' ) )
        self.assertTrue(hasattr(_lowerCamelCase    ,     '''current''' ) )
        self.assertTrue(hasattr(_lowerCamelCase    ,     '''total''' ) )

     with tempfile.TemporaryDirectory() as tmp_dir:
        lowerCAmelCase_			=      PyTorchBenchmarkArguments(
            models=[MODEL_ID]    ,     training=_lowerCamelCase    ,     inference=_lowerCamelCase    ,     sequence_lengths=[8]    ,     batch_sizes=[1]    ,     log_filename=os.path.join(_lowerCamelCase    ,     '''log.txt''' )    ,     log_print=_lowerCamelCase    ,     trace_memory_line_by_line=_lowerCamelCase    ,     multi_process=_lowerCamelCase    ,     )
        lowerCAmelCase_			=      PyTorchBenchmark(_lowerCamelCase )
        lowerCAmelCase_			=      benchmark.run()
        _check_summary_is_not_empty(result.inference_summary )
        _check_summary_is_not_empty(result.train_summary )
        self.assertTrue(Path(os.path.join(_lowerCamelCase    ,     '''log.txt''' ) ).exists() )





