Ejemplo n.º 1
0
  def benchmark_with_function_custom_loops_300_epochs_8_gpus(self):
    kwargs = utils.get_cifar10_kwargs()
    kwargs.update({'epochs': 300, 'data_format': 'channels_first',
                   'bottleneck': False, 'compression': 1., 'num_gpu': 8,
                   'batch_size': 512})

    self._run_and_report_benchmark(**kwargs)
    def benchmark_with_function_custom_loops_300_epochs_8_gpus(self):
        kwargs = utils.get_cifar10_kwargs()
        kwargs.update({
            'epochs': 300,
            'data_format': 'channels_first',
            'bottleneck': False,
            'compression': 1.,
            'num_gpu': 8,
            'batch_size': 512
        })

        self._run_and_report_benchmark(**kwargs)
 def benchmark_with_function_custom_loops(self):
     kwargs = utils.get_cifar10_kwargs()
     self._run_and_report_benchmark(**kwargs)
Ejemplo n.º 4
0
 def benchmark_with_keras_fit(self):
   kwargs = utils.get_cifar10_kwargs()
   kwargs.update({'train_mode': 'keras_fit'})
   self._run_and_report_benchmark(**kwargs)
Ejemplo n.º 5
0
 def benchmark_without_function_custom_loops(self):
   kwargs = utils.get_cifar10_kwargs()
   kwargs.update({'enable_function': False})
   self._run_and_report_benchmark(**kwargs)
Ejemplo n.º 6
0
 def benchmark_with_keras_fit_300_epochs(self):
   kwargs = utils.get_cifar10_kwargs()
   kwargs.update({'epochs': 300, 'data_format': 'channels_first',
                  'train_mode': 'keras_fit', 'bottleneck': False,
                  'compression': 1.})
   self._run_and_report_benchmark(**kwargs)
Ejemplo n.º 7
0
 def benchmark_with_function_custom_loops(self):
   kwargs = utils.get_cifar10_kwargs()
   self._run_and_report_benchmark(**kwargs)