tiny yolo not working

This commit is contained in:
Francesco Gatti
2017-08-07 15:05:48 +02:00
parent 34198a4e8d
commit 7d570c0df4
6 changed files with 112 additions and 14 deletions
+3 -2
View File
@@ -32,9 +32,10 @@ Network::~Network() {
value_type* Network::infer(dataDim_t &dim, value_type* data) {
//do infer for every layer
for(int i=0; i<num_layers; i++)
for(int i=0; i<num_layers; i++) {
data = layers[i]->infer(dim, data);
//dim.print();
}
checkCuda(cudaDeviceSynchronize());
return data;
}