Handle arguments were supported in V1.
(self, kwargs)
| 1466 | self._chief_worker_only = True |
| 1467 | |
| 1468 | def _validate_kwargs(self, kwargs): |
| 1469 | """Handle arguments were supported in V1.""" |
| 1470 | if kwargs.get('write_grads', False): |
| 1471 | logging.warning('`write_grads` will be ignored in TensorFlow 2.0 ' |
| 1472 | 'for the `TensorBoard` Callback.') |
| 1473 | if kwargs.get('batch_size', False): |
| 1474 | logging.warning('`batch_size` is no longer needed in the ' |
| 1475 | '`TensorBoard` Callback and will be ignored ' |
| 1476 | 'in TensorFlow 2.0.') |
| 1477 | if kwargs.get('embeddings_layer_names', False): |
| 1478 | logging.warning('`embeddings_layer_names` is not supported in ' |
| 1479 | 'TensorFlow 2.0. Instead, all `Embedding` layers ' |
| 1480 | 'will be visualized.') |
| 1481 | if kwargs.get('embeddings_data', False): |
| 1482 | logging.warning('`embeddings_data` is not supported in TensorFlow ' |
| 1483 | '2.0. Instead, all `Embedding` variables will be ' |
| 1484 | 'visualized.') |
| 1485 | |
| 1486 | unrecognized_kwargs = set(kwargs.keys()) - { |
| 1487 | 'write_grads', 'embeddings_layer_names', 'embeddings_data', 'batch_size' |
| 1488 | } |
| 1489 | |
| 1490 | # Only allow kwargs that were supported in V1. |
| 1491 | if unrecognized_kwargs: |
| 1492 | raise ValueError('Unrecognized arguments in `TensorBoard` ' |
| 1493 | 'Callback: ' + str(unrecognized_kwargs)) |
| 1494 | |
| 1495 | def set_model(self, model): |
| 1496 | """Sets Keras model and writes graph if specified.""" |