Merge pull request PaddlePaddle#10983 from panyx0718/fix_test

Fix test to use less gpu memory
huangjun12 · May 28, 2018 · 8075a11 · 8075a11
2 parents 25aa453 + c001961
commit 8075a11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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;