Searched refs:step_counter (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/examples/tf2_showcase/ |
| D | mnist.py | 140 def train(model, optimizer, dataset, step_counter, log_interval=None, argument
148 10, global_step=step_counter):
159 zip(grads, model.variables), global_step=step_counter)
211 step_counter = tf.train.get_or_create_global_step()
213 model=model, optimizer=optimizer, step_counter=step_counter)
221 train(model, optimizer, train_ds, step_counter,
226 step_counter.numpy(),
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
| D | mnist_test.py | 65 step_counter = tf.train.get_or_create_global_step()
81 burn_dataset, step_counter,
88 measure_dataset, step_counter,
|
| D | mnist.py | 198 discriminator_optimizer, dataset, step_counter, argument
218 tf.assign_add(step_counter, 1)
221 log_interval, global_step=step_counter):
307 checkpoint.step_counter.numpy(),
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dnn_linear_combined_test.py | 995 step_counter = _StepCounterHook()
998 classifier.fit(input_fn=input_fn, steps=100, monitors=[step_counter])
1000 self.assertEqual(100, step_counter.steps)
1014 step_counter = _StepCounterHook()
1019 classifier.fit(input_fn=input_fn, steps=100, monitors=[step_counter])
1021 self.assertEqual(100, step_counter.steps)
1037 step_counter = _StepCounterHook()
1044 classifier.fit(input_fn=input_fn, steps=100, monitors=[step_counter])
1051 self.assertIn(step_counter.steps, [50, 51])
1056 self.assertIn(step_counter.steps, [51, 52])
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | supervisor.py | 1046 def __init__(self, sv, sess, step_counter=None): argument
1060 step_counter = sv.global_step if step_counter is None else step_counter
1061 self._step_counter = step_counter
|