Commit 059ea3ba authored by William Chargin's avatar William Chargin Committed by TensorFlower Gardener
Browse files

Use V2 summary writer creation logic in Keras callback

A single `TensorBoard` Keras callback instance can now be used in
multiple calls to `fit` or `evaluate`. Prior to this change, the
underlying writer resource would be freed after the first call, causing
the second call to fail. The bulk of the heavy lifting was done in
cl/235249883; this change just hooks up the callback to the new
interface.

Fixes <#25707>.

Tested:
Regression test added; prior to this change, it fails with:

```
NotFoundError: Resource localhost/logdir:/<snip>/N10tensorflow22SummaryWriterInterfaceE does not exist. [Op:WriteGraphSummary]
```
PiperOrigin-RevId: 236047085
parent 1f691afc
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment