diff --git a/paddle/contrib/inference/test_paddle_inference_api_impl.cc b/paddle/contrib/inference/test_paddle_inference_api_impl.cc index 43b068fb42c5e5c58f2932c3e528fd0fa0502ec3..b103cfab3f09ede6bdcc9b0a36ccfa4d36ee9ffe 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;