Skip to content
This repository has been archived by the owner on Apr 19, 2023. It is now read-only.

Commit

Permalink
Merge pull request #167 from inferno-pytorch/constantinpape-patch-1
Browse files Browse the repository at this point in the history
Update tensorboard.py
  • Loading branch information
constantinpape committed Mar 12, 2019
2 parents 50fa978 + eb83a77 commit 4bb6bc3
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions inferno/trainers/callbacks/logging/tensorboard.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,9 @@ def __init__(self, log_directory=None,
log_scalars_every : str or tuple or inferno.utils.train_utils.Frequency
How often scalars should be logged to Tensorboard. By default, once every iteration.
log_images_every : str or tuple or inferno.utils.train_utils.Frequency
How often images should be logged to Tensorboard. By default, once every iteration.
How often images should be logged to Tensorboard. By default, once every epoch.
log_histograms_every : str or tuple or inferno.utils.train_utils.Frequency
How often histograms should be logged to Tensorboard. By default, once every iteration.
How often histograms should be logged to Tensorboard. By default, never.
send_image_at_batch_indices : list or str
The indices of the batches to be logged. An `image_batch` usually has the shape
(num_samples, num_channels, num_rows, num_cols). By setting this argument to say
Expand Down Expand Up @@ -108,7 +108,7 @@ def log_scalars_now(self):
@property
def log_images_every(self):
if self._log_images_every is None:
self._log_images_every = tru.Frequency(1, 'iterations')
self._log_images_every = tru.Frequency(1, 'epochs')
return self._log_images_every

@log_images_every.setter
Expand All @@ -126,7 +126,7 @@ def log_images_now(self):
@property
def log_histograms_every(self):
if self._log_histograms_every is None:
self._log_histograms_every = tru.Frequency(1, 'iterations')
self._log_histograms_every = tru.Frequency('never')
return self._log_histograms_every

@log_histograms_every.setter
Expand Down

0 comments on commit 4bb6bc3

Please sign in to comment.