Merge pull request #42583 from raven38:raven38-patch
PiperOrigin-RevId: 328273735 Change-Id: I57b78630da1dd4482114fdba7a2a6e2493d83b08
This commit is contained in:
commit
2856a636b1
@ -138,7 +138,7 @@ class _DataSet(object):
|
||||
Args:
|
||||
images: The images
|
||||
labels: The labels
|
||||
fake_data: Ignore inages and labels, use fake data.
|
||||
fake_data: Ignore images and labels, use fake data.
|
||||
one_hot: Bool, return the labels as one hot vectors (if True) or ints (if
|
||||
False).
|
||||
dtype: Output image dtype. One of [uint8, float32]. `uint8` output has
|
||||
@ -330,4 +330,3 @@ def read_data_sets(train_dir,
|
||||
test = _DataSet(test_images, test_labels, **options)
|
||||
|
||||
return _Datasets(train=train, validation=validation, test=test)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user