GitHub Issue #40462

PiperOrigin-RevId: 317400375
Change-Id: I13891b5c2f41ac97674ddfda679c4273d53b25ef
This commit is contained in:
A. Unique TensorFlower 2020-06-19 16:23:33 -07:00 committed by TensorFlower Gardener
parent fac99746cb
commit a70ad66828

View File

@ -147,7 +147,7 @@ def image_dataset_from_directory(directory,
'directory. If you wish to infer the labels from the subdirectory '
'names in the target directory, pass `labels="inferred"`. '
'If you wish to get a dataset that only contains images '
'(no labels), pass `labels=None`.')
'(no labels), pass `labels_mode=None`.')
if class_names:
raise ValueError('You can only pass `class_names` if the labels are '
'inferred from the subdirectory names in the target '