def testNoStrategyEagerCpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='off', num_gpus=0) self._run(strategy, tf_function=False)
def testEagerCpu(self): print('TF 2.0 enable:{}'.format(tf2.enabled())) strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='default', num_gpus=0) self._run(strategy, tf_function=False)
def testNoStrategyCpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='off', num_gpus=0) self._run(strategy)
def benchmark_dqn_8_gpu_xla(self): utils.set_session_config(enable_xla=True) strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='default', num_gpus=8) self._run(strategy, batch_size=64 * 8)
def testCpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='default', num_gpus=0) self._run(strategy)
def benchmark_dqn_mirrored_1_gpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='mirrored', num_gpus=1) self._run(strategy)
def benchmark_dqn_eagerly_1_gpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='default', num_gpus=1) self._run(strategy, tf_function=False)
def benchmark_dqn_8_gpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='default', num_gpus=8) self._run(strategy, batch_size=64 * 8)
def benchmark_dqn_1_gpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='default', num_gpus=1) self._run(strategy)
def benchmark_dqn_no_dist_strat_1_gpu_xla(self): utils.set_session_config(enable_xla=True) strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='off', num_gpus=1) self._run(strategy)
def benchmark_dqn_no_dist_strat_1_gpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='off', num_gpus=1) self._run(strategy)
def benchmark_dqn_no_dist_strat_eagerly_cpu(self): strategy = distribution_strategy_utils.get_distribution_strategy( distribution_strategy='off', num_gpus=0) self._run(strategy, tf_function=False)