Mnist works at the moment with trt8,others like yolo4tiny and mobilenet generate the engine files but crash after throwing nvifer1::CudaRuntimeError and when demo is being run ,it doesnt deserialize properly and crashes

2021-08-29 03:18:58 +05:30
parent 3d8b1ac494
commit 2ffe07057e
12 changed files with 255 additions and 42 deletions
@@ -128,7 +128,7 @@ int main() {
 	builder->setMaxBatchSize(1);
 	config->setMaxWorkspaceSize(1 << 20);

-	auto engine = builder->buildCudaEngine(*network);
+	auto engine = builder->buildEngineWithConfig(*network,*config);
 	// we don't need the network any more
 	network->destroy();