From c0019610346cccbc72f19adc431649b1844671f9 Mon Sep 17 00:00:00 2001 From: Xin Pan Date: Mon, 28 May 2018 23:05:11 +0800 Subject: [PATCH] Fix test to use less gpu memory --- paddle/contrib/inference/test_paddle_inference_api_impl.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/contrib/inference/test_paddle_inference_api_impl.cc b/paddle/contrib/inference/test_paddle_inference_api_impl.cc index 43b068fb42..b103cfab3f 100644 --- a/paddle/contrib/inference/test_paddle_inference_api_impl.cc +++ b/paddle/contrib/inference/test_paddle_inference_api_impl.cc @@ -44,7 +44,7 @@ TEST(paddle_inference_api_impl, word2vec) { VisConfig config; config.model_dir = FLAGS_dirname + "word2vec.inference.model"; LOG(INFO) << "dirname " << config.model_dir; - config.fraction_of_gpu_memory = 0.85; + config.fraction_of_gpu_memory = 0.15; config.device = 0; config.share_variables = true; -- GitLab