(self,
log_dir='logs',
histogram_freq=0,
write_graph=True,
write_images=False,
update_freq='epoch',
profile_batch=2,
embeddings_freq=0,
embeddings_metadata=None,
**kwargs)
| 1422 | # pylint: enable=line-too-long |
| 1423 | |
| 1424 | def __init__(self, |
| 1425 | log_dir='logs', |
| 1426 | histogram_freq=0, |
| 1427 | write_graph=True, |
| 1428 | write_images=False, |
| 1429 | update_freq='epoch', |
| 1430 | profile_batch=2, |
| 1431 | embeddings_freq=0, |
| 1432 | embeddings_metadata=None, |
| 1433 | **kwargs): |
| 1434 | super(TensorBoard, self).__init__() |
| 1435 | self._validate_kwargs(kwargs) |
| 1436 | |
| 1437 | self.log_dir = log_dir |
| 1438 | self.histogram_freq = histogram_freq |
| 1439 | self.write_graph = write_graph |
| 1440 | self.write_images = write_images |
| 1441 | if update_freq == 'batch': |
| 1442 | self.update_freq = 1 |
| 1443 | else: |
| 1444 | self.update_freq = update_freq |
| 1445 | self.embeddings_freq = embeddings_freq |
| 1446 | self.embeddings_metadata = embeddings_metadata |
| 1447 | |
| 1448 | self._samples_seen = 0 |
| 1449 | self._samples_seen_at_last_write = 0 |
| 1450 | self._current_batch = 0 |
| 1451 | |
| 1452 | # A collection of file writers currently in use, to be closed when |
| 1453 | # training ends for this callback. Writers are keyed by the |
| 1454 | # directory name under the root logdir: e.g., "train" or |
| 1455 | # "validation". |
| 1456 | self._train_run_name = 'train' |
| 1457 | self._validation_run_name = 'validation' |
| 1458 | self._writers = {} |
| 1459 | |
| 1460 | self._profile_batch = profile_batch |
| 1461 | # True when a trace is running. |
| 1462 | self._is_tracing = False |
| 1463 | |
| 1464 | # TensorBoard should only write summaries on the chief when in a |
| 1465 | # Multi-Worker setting. |
| 1466 | self._chief_worker_only = True |
| 1467 | |
| 1468 | def _validate_kwargs(self, kwargs): |
| 1469 | """Handle arguments were supported in V1.""" |
nothing calls this directly
no test coverage detected