From 096b9137e5c9f8406e8c2c9a81c0d12ddada24bc Mon Sep 17 00:00:00 2001 From: gmcather Date: Wed, 11 Apr 2018 12:44:01 +0000 Subject: [PATCH] fix infer.py --- fluid/text_classification/infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/text_classification/infer.py b/fluid/text_classification/infer.py index 6b39f92e..b377a053 100644 --- a/fluid/text_classification/infer.py +++ b/fluid/text_classification/infer.py @@ -46,6 +46,6 @@ if __name__ == "__main__": batch_size = 128, buf_size = 50000) model_path = sys.argv[1] - test(test_reader, use_cuda=True, + infer(test_reader, use_cuda=False, save_dirname=model_path) -- GitLab