OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

TPUEstimator: Fix the outfeed thread join.

Browse Source 
PiperOrigin-RevId: 165651781
This commit is contained in:
Jonathan Hseu 2017-08-17 17:26:07 -07:00 committed by TensorFlower Gardener
parent 19a55725af
commit e31346452d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/tpu/python/tpu/tpu_estimator.py
View File

@ -283,7 +283,7 @@ class TPUInfeedOutfeedSessionHook(session_run_hook.SessionRunHook):
if self._dequeue_ops is not None: if self._dequeue_ops is not None:
logging.info('Stop output thread controller') logging.info('Stop output thread controller')
self._infeed_thd_controller.join() self._outfeed_thd_controller.join()
logging.info('Shutdown TPU system.') logging.info('Shutdown TPU system.')
session.run(self._finalize_op) session.run(self._finalize_op)