Skip to content

Commit c001961

Browse files
committed
Fix test to use less gpu memory
1 parent 7323be1 commit c001961

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/contrib/inference/test_paddle_inference_api_impl.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ TEST(paddle_inference_api_impl, word2vec) {
4444
VisConfig config;
4545
config.model_dir = FLAGS_dirname + "word2vec.inference.model";
4646
LOG(INFO) << "dirname " << config.model_dir;
47-
config.fraction_of_gpu_memory = 0.85;
47+
config.fraction_of_gpu_memory = 0.15;
4848
config.device = 0;
4949
config.share_variables = true;
5050

0 commit comments

Comments
 (0)