Skip to content

Commit 320c78e

Browse files
committed
fix commets
test=develop
1 parent efa5bac commit 320c78e

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

paddle/fluid/inference/api/demo_ci/trt_mobilenet_demo.cc

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,15 +18,9 @@ limitations under the License. */
1818

1919
#include <gflags/gflags.h>
2020
#include <glog/logging.h> // use glog instead of CHECK to avoid importing other paddle header files.
21-
#include <fstream>
22-
#include <iostream>
23-
24-
// #include "paddle/fluid/platform/enforce.h"
2521
#include "paddle/fluid/inference/demo_ci/utils.h"
2622

27-
#ifdef PADDLE_WITH_CUDA
2823
DECLARE_double(fraction_of_gpu_memory_to_use);
29-
#endif
3024
DEFINE_string(modeldir, "", "Directory of the inference model.");
3125
DEFINE_string(refer, "", "path to reference result for comparison.");
3226
DEFINE_string(
@@ -38,7 +32,7 @@ namespace paddle {
3832
namespace demo {
3933

4034
/*
41-
* Use the native fluid engine to inference the demo.
35+
* Use the tensorrt fluid engine to inference the demo.
4236
*/
4337
void Main() {
4438
std::unique_ptr<PaddlePredictor> predictor;

0 commit comments

Comments
 (0)