Yolov3_tiny works on tensorRT
Signed-off-by: fbagni <gattinomicino>
This commit is contained in:
+2
-2
@@ -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) );
|
||||
|
||||
Reference in New Issue
Block a user