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

Remove obsolete TODOs.

Browse Source 
PiperOrigin-RevId: 254039929
This commit is contained in:
Shining Sun 2019-06-19 11:45:13 -07:00 committed by TensorFlower Gardener
parent f4e64ca17c
commit b8ad028165
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/distribute/multi_worker_callback_test.py
View File

@ -146,16 +146,12 @@ class KerasMultiWorkerCallbackTest(test_base.IndependentWorkerTestBase,
**kwargs): **kwargs):
extension = os.path.splitext(saving_filepath)[1] extension = os.path.splitext(saving_filepath)[1]
# TODO(rchao): Remove using .h5 once b/134551335 is fixed.
extension = '.h5'
# Incorporate type/index information and thread id in saving_filepath to # Incorporate type/index information and thread id in saving_filepath to
# ensure every worker has a unique path. Note that in normal use case the # ensure every worker has a unique path. Note that in normal use case the
# saving_filepath will be the same for all workers, but we use different # saving_filepath will be the same for all workers, but we use different
# ones here just to test out chief saves checkpoint but non-chief doesn't. # ones here just to test out chief saves checkpoint but non-chief doesn't.
# TODO(b/134551335): Must save to hdf5 until bug with copying
# MirroredVariables is resolved.
saving_filepath = os.path.join( saving_filepath = os.path.join(
test_obj.get_temp_dir(), 'checkpoint_%s_%d%s' % test_obj.get_temp_dir(), 'checkpoint_%s_%d%s' %
(test_base.get_task_type(), test_base.get_task_index(), extension)) (test_base.get_task_type(), test_base.get_task_index(), extension))