diff options
Diffstat (limited to 'tensorflow/examples/how_tos/reading_data/convert_to_records.py')
| -rw-r--r-- | tensorflow/examples/how_tos/reading_data/convert_to_records.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/examples/how_tos/reading_data/convert_to_records.py b/tensorflow/examples/how_tos/reading_data/convert_to_records.py index c8819e05ce..da42b5cc38 100644 --- a/tensorflow/examples/how_tos/reading_data/convert_to_records.py +++ b/tensorflow/examples/how_tos/reading_data/convert_to_records.py @@ -68,6 +68,7 @@ def convert_to(images, labels, name): 'label': _int64_feature(int(labels[index])), 'image_raw': _bytes_feature(image_raw)})) writer.write(example.SerializeToString()) + writer.close() def main(argv): |