From 1e8a5f46a246fa5f7b448a1dbcc87a47e0e27660 Mon Sep 17 00:00:00 2001 From: frankwhzhang Date: Wed, 12 Dec 2018 14:24:14 +0800 Subject: [PATCH] fix python3 xrange bug, gru4rec/tagspace --- fluid/PaddleRec/gru4rec/infer.py | 2 +- fluid/PaddleRec/tagspace/infer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/PaddleRec/gru4rec/infer.py b/fluid/PaddleRec/gru4rec/infer.py index ccb2ce09..a39eb5db 100644 --- a/fluid/PaddleRec/gru4rec/infer.py +++ b/fluid/PaddleRec/gru4rec/infer.py @@ -78,6 +78,6 @@ if __name__ == "__main__": test_dir, "", batch_size=batch_size, buffer_size=1000, word_freq_threshold=0, is_train=False) - for epoch in xrange(start_index, last_index + 1): + for epoch in range(start_index, last_index + 1): epoch_path = model_dir + "/epoch_" + str(epoch) infer(test_reader=test_reader, use_cuda=use_cuda, model_path=epoch_path) diff --git a/fluid/PaddleRec/tagspace/infer.py b/fluid/PaddleRec/tagspace/infer.py index 2ecdaa5a..8cecbab6 100644 --- a/fluid/PaddleRec/tagspace/infer.py +++ b/fluid/PaddleRec/tagspace/infer.py @@ -80,6 +80,6 @@ if __name__ == "__main__": test_dir, "", vocab_tag_path, batch_size=1, neg_size=0, buffer_size=1000, is_train=False) - for epoch in xrange(start_index, last_index + 1): + for epoch in range(start_index, last_index + 1): epoch_path = model_dir + "/epoch_" + str(epoch) infer(test_reader=test_reader, vocab_tag=vocab_tag, use_cuda=False, model_path=epoch_path) -- GitLab