Unverified Commit 12e977e4 authored by Lukasz Kaiser's avatar Lukasz Kaiser Committed by GitHub
Browse files

Merge pull request #4417 from aarorasimbulus/patch-1

range doesn't return list anymore
parents 2eeb85fe c15aedf3
...@@ -125,7 +125,7 @@ def convert_data(trainingdata_dir, ...@@ -125,7 +125,7 @@ def convert_data(trainingdata_dir,
tf.python_io.TFRecordWriter("%s-%05i-of-%05i" % (output_file, i, tf.python_io.TFRecordWriter("%s-%05i-of-%05i" % (output_file, i,
output_shards))) output_shards)))
reading_order = range(len(file_handles)) * observations_per_class reading_order = list(range(len(file_handles))) * observations_per_class
random.shuffle(reading_order) random.shuffle(reading_order)
for c in reading_order: for c in reading_order:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment