yolo layers
This commit is contained in:
@@ -35,4 +35,4 @@ void activationELUForward(value_type* srcData, value_type* dstData, int size)
|
||||
|
||||
activation_elu<<<blocks, threads>>>(srcData, dstData, size);
|
||||
checkCuda( cudaDeviceSynchronize() );
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user