Skip to content

Commit

Permalink
Update model.cpp
Browse files Browse the repository at this point in the history
  • Loading branch information
lindsayshuo authored Apr 17, 2024
1 parent 20ae8f4 commit 6ce06d5
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions yolov8/src/model.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -622,9 +622,6 @@ nvinfer1::IHostMemory* buildEngineYolov8DetP6(nvinfer1::IBuilder* builder, nvinf
nvinfer1::IHostMemory* buildEngineYolov8DetP2(nvinfer1::IBuilder* builder, nvinfer1::IBuilderConfig* config,
nvinfer1::DataType dt, const std::string& wts_path, float& gd, float& gw,
int& max_channels) {

std::cout << "buildEngineYolov8DetP2 " << std::endl;

std::map<std::string, nvinfer1::Weights> weightMap = loadWeights(wts_path);
nvinfer1::INetworkDefinition* network = builder->createNetworkV2(0U);

Expand Down

0 comments on commit 6ce06d5

Please sign in to comment.