def test_one_epoch(self):
num_examples = 10
buffer_size = 10
batch_size = 1
embedding_dim = 4
enc_units = 4
dec_units = 4
epochs = 1
train.main(epochs, True, buffer_size, batch_size, 'datasets',
num_examples, embedding_dim, enc_units, dec_units)
def test_one_epoch(self):
num_examples = 10
buffer_size = 10
batch_size = 1
embedding_dim = 4
enc_units = 4
dec_units = 4
epochs = 1
train.main(epochs, True, buffer_size, batch_size, 'datasets', num_examples,
embedding_dim, enc_units, dec_units)
def _run_and_report_benchmark(self, **kwargs):
start_time_sec = time.time()
train_loss, test_loss = train.main(**kwargs)
wall_time_sec = time.time() - start_time_sec
extras = {'train_loss': train_loss, 'test_loss': test_loss}
self.report_benchmark(wall_time=wall_time_sec, extras=extras)
def _run_and_report_benchmark(self, **kwargs):
start_time_sec = time.time()
train_loss, test_loss = train.main(**kwargs)
wall_time_sec = time.time() - start_time_sec
extras = {'train_loss': train_loss,
'test_loss': test_loss}
self.report_benchmark(
wall_time=wall_time_sec, extras=extras)