try:
with tf.gfile.Open(trainable_labels_file) as tl:
for label in tl:
trainable_labels.add(label.strip())
except tf.errors.NotFoundError:
raise InvalidDataDirectory(
"Missing label file "{}" from data_dir".format(
After Change
// Skip header
next(reader, None)
for line in reader:
trainable_labels.add(line[2])
except tf.errors.NotFoundError:
raise InvalidDataDirectory(
"Missing label file "{}" from data_dir".format(