diff --git a/fluid/PaddleRec/gru4rec/infer.py b/fluid/PaddleRec/gru4rec/infer.py index ccb2ce097f10798b9e68c623a921b3e5bebf9af3..a39eb5db1607e661923ed08fba93e0606e9a1750 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 2ecdaa5a792da24495ee4b2c5ce1ab605e187ba9..8cecbab68d2285b50036cab678e59059689f7d65 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)