memory release

2020-06-02 12:43:06 +02:00
parent 0458f361b1
commit 62e4a3f779
7 changed files with 20 additions and 5 deletions
@@ -134,6 +134,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
        networkRT->markOutput(*input);

        std::cout<<"Selected maxBatchSize: "<<builderRT->getMaxBatchSize()<<"\n";
+        printCudaMemUsage();
        std::cout<<"Building tensorRT cuda engine...\n";
 #if NV_TENSORRT_MAJOR >= 6                
        engineRT = builderRT->buildEngineWithConfig(*networkRT, *configRT);