darknet parser tested

2020-06-01 14:55:08 +02:00
parent d2e2669b6d
commit 18794d52c2
21 changed files with 112 additions and 30 deletions
@@ -59,12 +59,16 @@ Network::Network(dataDim_t input_dim) {
 }

 Network::~Network() {
-    for(int i=0; i<num_layers; i++)
-        delete layers[i];
    checkCUDNN( cudnnDestroy(cudnnHandle) );
    checkERROR( cublasDestroy(cublasHandle) );
 }

+void Network::releaseLayers() {
+    for(int i=0; i<num_layers; i++)
+        delete layers[i];
+    num_layers = 0;
+}
+
 dnnType* Network::infer(dataDim_t &dim, dnnType* data) {

    //do infer for every layer