提交 096b9137 编写于 作者: G gmcather

fix infer.py

上级 d2b7bac9
......@@ -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)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册