experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Replace tf.keras.callbacks.TensorBoard callback with a new callback TensorBoardWithValidation inherited from the former callback, which makes evaluation at the end of specified epochs and export the results to tensorboard.

Browse Source 
PiperOrigin-RevId: 219749605
This commit is contained in:
Jing Li 2018-11-01 21:57:59 -07:00 committed by TensorFlower Gardener
parent a61fff3e12
commit a2f2c7b035
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/contrib/tpu/python/tpu/keras_support.py
View File

@ -1536,10 +1536,17 @@ class KerasTPUModel(models.Model):
verbose=1, verbose=1,
sample_weight=None, sample_weight=None,
steps=None): steps=None):
assert not self._numpy_to_infeed_manager_list # Ensure empty. original_numpy_to_infeed_manager_list = []
if self._numpy_to_infeed_manager_list:
# evaluate call may be executed as callbacks during the training. In this
# case, _numpy_to_infeed_manager_list is not empty, so save it for
# recovery at the end of evaluate call.
original_numpy_to_infeed_manager_list = self._numpy_to_infeed_manager_list
self._numpy_to_infeed_manager_list = []
with _tpu_session_context(): with _tpu_session_context():
infeed_managers = [] # Managers to clean up at the end of the fit call. # Managers to clean up at the end of the evaluate call.
infeed_managers = []
if isinstance(x, dataset_ops.Dataset): if isinstance(x, dataset_ops.Dataset):
# TODO(b/111413240): Support taking a tf.data.Dataset directly. # TODO(b/111413240): Support taking a tf.data.Dataset directly.
raise ValueError( raise ValueError(
@ -1569,7 +1576,8 @@ class KerasTPUModel(models.Model):
return super(KerasTPUModel, self).evaluate(x, y, batch_size, verbose, return super(KerasTPUModel, self).evaluate(x, y, batch_size, verbose,
sample_weight, steps) sample_weight, steps)
finally: finally:
self._numpy_to_infeed_manager_list = [] self._numpy_to_infeed_manager_list = (
original_numpy_to_infeed_manager_list)
def _pipeline_fit(self, x, y, batch_size, epochs, verbose, callbacks, def _pipeline_fit(self, x, y, batch_size, epochs, verbose, callbacks,
validation_split, validation_data, shuffle, class_weight, validation_split, validation_data, shuffle, class_weight,