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

Re-introduce support for key-value records into tf.data.read_batch_features.

Browse Source 
PiperOrigin-RevId: 175580235
This commit is contained in:
Jiri Simsa 2017-11-13 13:36:50 -08:00 committed by TensorFlower Gardener
parent bac56b37be
commit 9642c1c016
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/data/python/ops/readers.py
View File

@ -171,6 +171,8 @@ def read_batch_features(file_pattern,
dataset = reader(filenames, *reader_args)
else:
dataset = reader(filenames)
if dataset.output_types == (dtypes.string, dtypes.string):
dataset = dataset.map(lambda _, v: v)
if num_epochs != 1:
dataset = dataset.repeat(num_epochs)
if randomize_input: