Skip to content
This repository has been archived by the owner on Dec 16, 2022. It is now read-only.

Commit

Permalink
Make per-batch logging quieter. (#3020)
Browse files Browse the repository at this point in the history
- Fixes #2998
  • Loading branch information
brendan-ai2 committed Jul 1, 2019
1 parent 57870f5 commit ae72049
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion allennlp/training/callbacks/log_to_tensorboard.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def batch_end_logging(self, trainer: 'CallbackTrainer'):
self.cumulative_batch_size += cur_batch
if (trainer.batches_this_epoch - 1) % self.log_batch_size_period == 0:
average = self.cumulative_batch_size / trainer.batches_this_epoch
logger.info(f"current batch size: {cur_batch} mean batch size: {average}")
logger.debug(f"current batch size: {cur_batch} mean batch size: {average}")
self.tensorboard.add_train_scalar("current_batch_size", cur_batch)
self.tensorboard.add_train_scalar("mean_batch_size", average)

Expand Down

0 comments on commit ae72049

Please sign in to comment.