Yolov3_tiny works on tensorRT

Signed-off-by: fbagni <gattinomicino>
This commit is contained in:
fbagni
2019-12-23 16:11:41 +01:00
parent f0ea2c027f
commit 7e7f480e2b
5 changed files with 20 additions and 32 deletions
+2 -2
View File
@@ -17,8 +17,8 @@ void Conv2d::initCUDNN(bool back) {
idim = output_dim;
odim = input_dim;
}
idim.print();
odim.print();
//idim.print();
//odim.print();
checkCUDNN( cudnnCreateFilterDescriptor(&filterDesc) );
checkCUDNN( cudnnCreateConvolutionDescriptor(&convDesc) );