Skip to content
Snippets Groups Projects
Commit 9c5df446 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Solving python3 issue

parent 046ac8d8
Branches
Tags
1 merge request!24Implemented random triplet batching
Pipeline #
Hide whitespace changes
Inline Side-by-side
bob/learn/tensorflow/utils/util.py
+ 1
− 1
View file @ 9c5df446
...@@ -62,7 +62,7 @@ def create_mnist_tfrecord(tfrecords_filename, data, labels, n_samples=6000): ...@@ -62,7 +62,7 @@ def create_mnist_tfrecord(tfrecords_filename, data, labels, n_samples=6000):
img_raw = img.tostring() img_raw = img.tostring()
feature = {'data': _bytes_feature(img_raw), feature = {'data': _bytes_feature(img_raw),
'label': _int64_feature(labels[i]), 'label': _int64_feature(labels[i]),
'key': _bytes_feature("") 'key': _bytes_feature(b'-')
} }
example = tf.train.Example(features=tf.train.Features(feature=feature)) example = tf.train.Example(features=tf.train.Features(feature=feature))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment