Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 f277f53c 编写于 作者: N nhzlx
浏览文件
操作

out of memory... i bet it's the last time commit for this pr

上级 a0a67893
隐藏空白更改
内联 并排
Showing with 3 addition and 3 deletion
paddle/fluid/inference/tests/api/trt_models_tester.cc
浏览文件 @ f277f53c
...@@ -26,7 +26,7 @@ NativeConfig GetConfigNative() { ...@@ -26,7 +26,7 @@ NativeConfig GetConfigNative() {
NativeConfig config; NativeConfig config;
config.model_dir = FLAGS_dirname; config.model_dir = FLAGS_dirname;
// LOG(INFO) << "dirname " << config.model_dir; // LOG(INFO) << "dirname " << config.model_dir;
config.fraction_of_gpu_memory = 0.7; config.fraction_of_gpu_memory = 0.45;
config.use_gpu = true; config.use_gpu = true;
config.device = 0; config.device = 0;
return config; return config;
...@@ -36,7 +36,7 @@ TensorRTConfig GetConfigTRT() { ...@@ -36,7 +36,7 @@ TensorRTConfig GetConfigTRT() {
TensorRTConfig config; TensorRTConfig config;
config.model_dir = FLAGS_dirname; config.model_dir = FLAGS_dirname;
config.use_gpu = true; config.use_gpu = true;
config.fraction_of_gpu_memory = 0.1; config.fraction_of_gpu_memory = 0.2;
config.device = 0; config.device = 0;
config.max_batch_size = 3; config.max_batch_size = 3;
return config; return config;
...@@ -99,7 +99,7 @@ TEST(trt_models_test, main) { ...@@ -99,7 +99,7 @@ TEST(trt_models_test, main) {
std::vector<std::string> infer_models = {"mobilenet", "resnet50", std::vector<std::string> infer_models = {"mobilenet", "resnet50",
"resnext50"}; "resnext50"};
for (auto &model_dir : infer_models) { for (auto &model_dir : infer_models) {
CompareTensorRTWithFluid(5, FLAGS_dirname + "/" + model_dir); CompareTensorRTWithFluid(1, FLAGS_dirname + "/" + model_dir);
} }
} }
} // namespace paddle } // namespace paddle
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或