log interval 1 when benchmark

pull/837/head
Hui Zhang 4 years ago
parent f15e1ff732
commit 576e94da04

@ -100,7 +100,7 @@ class DeepSpeech2Trainer(Trainer):
iteration_time = time.time() - start
msg += "train time: {:>.3f}s, ".format(iteration_time)
msg += "batch cost: {:>.3f}s, ".format(iteration_time)
msg += "batch size: {}, ".format(self.config.collator.batch_size)
msg += "accum: {}, ".format(train_conf.accum_grad)
msg += ', '.join('{}: {:>.6f}'.format(k, v)

@ -106,6 +106,7 @@ class Trainer():
if self.args.benchmark_batch_size:
with UpdateConfig(self.config):
self.config.collator.batch_size = self.args.benchmark_batch_size
self.config.training.log_interval = 1
logger.info(
f"Benchmark reset batch-size: {self.args.benchmark_batch_size}")

Loading…
Cancel
Save