Exemplos de create_global_step em Python

Exemplo n.º 1

Arquivo: standard_runner_test.py Projeto: Brumana-Gregis-Rota/tesi

 def __init__(self, options=None):
     self.strategy = tf.distribute.get_strategy()
     self.global_step = utils.create_global_step()
     distribute = self.strategy.experimental_distribute_datasets_from_function
     dataset = distribute(dataset_fn)
     super().__init__(train_dataset=dataset, options=options)

Exemplo n.º 2

 def __init__(self, options=None):
     self.strategy = tf.distribute.get_strategy()
     self.global_step = utils.create_global_step()
     dataset = self.strategy.distribute_datasets_from_function(dataset_fn)
     super().__init__(eval_dataset=dataset, options=options)

Exemplo n.º 3

Arquivo: utils_test.py Projeto: Haimzis/model

 def test_create_global_step(self):
     step = utils.create_global_step()
     self.assertEqual(step.dtype, tf.int64)
     self.assertEqual(step, 0)
     step.assign_add(1)
     self.assertEqual(step, 1)