NetworkRT (deallocations to be done)

2017-08-03 12:16:57 +02:00
parent e8355cee67
commit 4526e2767a
11 changed files with 344 additions and 34 deletions
@@ -37,6 +37,7 @@ value_type* Network::infer(dataDim_t &dim, value_type* data) {
    for(int i=0; i<num_layers; i++)
        data = layers[i]->infer(dim, data);

+    checkCuda(cudaDeviceSynchronize());
    return data;
 }