"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "28bd85d1d87707a0b0978407f9b2f2f13ae0a333"
Unverified Commit c15aedf3 authored by Akshit Arora's avatar Akshit Arora Committed by GitHub
Browse files

range doesn't return list anymore

Code was written for python2
parent 3d2f8959
...@@ -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