X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;ds=sidebyside;f=voxel_flow_train.py;h=4440e798410e9ec4b16f638146a5d4fcf8862be4;hb=bed00c52fa6fbd9bda76f0f4d2fa2e7c0ff08922;hp=f6d3abb67a0f062e6750c0b3bd8471174be1b8f6;hpb=c74057bc77fb9bc4eb75668896adfb85ddb99251;p=voxel-flow

diff --git a/voxel_flow_train.py b/voxel_flow_train.py
index f6d3abb..4440e79 100755
--- a/voxel_flow_train.py
+++ b/voxel_flow_train.py
@@ -118,7 +118,7 @@ def train(dataset_frame1, dataset_frame2, dataset_frame3):
     # load_fn_frame3 = partial(dataset_frame3.process_func)
     # p_queue_frame3 = PrefetchQueue(load_fn_frame3, data_list_frame3, FLAGS.batch_size, shuffle=False, num_workers=num_workers)
 
-    for step in xrange(0, FLAGS.max_steps):
+    for step in range(0, FLAGS.max_steps):
       batch_idx = step % epoch_num
       
       batch_data_list_frame1 = data_list_frame1[int(batch_idx * FLAGS.batch_size) : int((batch_idx + 1) * FLAGS.batch_size)]