Fix test to use less gpu memory

c0019610 · Xin Pan · 7323be17 · c0019610
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/contrib/inference/test_paddle_inference_api_impl.cc paddle/contrib/inference/test_paddle_inference_api_impl.cc +1 -1

未找到文件。
--- 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;