update

juzigithub · Dec 10, 2021 · 6a41326 · 6a41326
1 parent 5ca485f
commit 6a41326
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/application/app_yolo.cpp b/src/application/app_yolo.cpp
@@ -136,7 +136,7 @@ static void test(Yolo::Type type, TRT::Mode mode, const string& model){
 
     string onnx_file = iLogger::format("%s.onnx", name);
     string model_file = iLogger::format("%s.%s.trtmodel", name, mode_name);
-    int test_batch_size = 3;
+    int test_batch_size = 16;
 
     if(not iLogger::exists(model_file)){
         TRT::compile(
@@ -174,7 +174,7 @@ void multi_gpu_test(){
 
 int app_yolo(){
 
-    test(Yolo::Type::V5, TRT::Mode::FP32, "yolov5s");
+    test(Yolo::Type::V5, TRT::Mode::INT8, "yolov5s");
     //test(Yolo::Type::V3, TRT::Mode::FP32, "yolov3");
 
     // multi_gpu_test();

diff --git a/src/tensorRT/builder/trt_builder.cpp b/src/tensorRT/builder/trt_builder.cpp
@@ -498,7 +498,6 @@ namespace TRT {
 		if(source.type() == ModelSourceType::OnnX || source.type() == ModelSourceType::OnnXData){
 
 			const auto explicitBatch = 1U << static_cast<uint32_t>(nvinfer1::NetworkDefinitionCreationFlag::kEXPLICIT_BATCH);
-			//network = shared_ptr<INetworkDefinition>(builder->createNetworkV2(explicitBatch), destroy_nvidia_pointer<INetworkDefinition>);
 			network = shared_ptr<INetworkDefinition>(builder->createNetworkV2(explicitBatch), destroy_nvidia_pointer<INetworkDefinition>);
 
 			vector<nvinfer1::Dims> dims_setup(inputsDimsSetup.size());