tiny yolo not working
This commit is contained in:
+3
-2
@@ -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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user