Searched refs:train_one_epoch (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/contrib/eager/python/examples/mnist/ |
D | mnist_test.py | 42 def train_one_epoch(defun=False): function 50 mnist.train_one_epoch(model, optimizer, dataset) 66 train_one_epoch(defun=False) 72 train_one_epoch(defun=True)
|
D | mnist.py | 119 def train_one_epoch(model, optimizer, dataset, log_interval=None): function 200 train_one_epoch(model, optimizer, train_ds, FLAGS.log_interval)
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
D | mnist_test.py | 79 mnist.train_one_epoch(generator, discriminator, generator_optimizer, 85 mnist.train_one_epoch(generator, discriminator, generator_optimizer,
|
D | mnist.py | 201 def train_one_epoch(generator, discriminator, function 296 train_one_epoch(generator, discriminator, generator_optimizer,
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_colorbot/ |
D | rnn_colorbot.py | 201 def train_one_epoch(model, optimizer, train_data, log_interval=10): function 258 train_one_epoch(model, optimizer, train_data, FLAGS.log_interval)
|
D | rnn_colorbot_test.py | 57 rnn_colorbot.train_one_epoch(model, optimizer, dataset)
|