Batchnorm eps fix, works on jetpack 4.3

This commit is contained in:
xavier
2020-01-15 18:06:02 +01:00
parent 7233b065a8
commit 33844c1ab2
5 changed files with 21 additions and 17 deletions
+4 -3
View File
@@ -3,9 +3,10 @@ tkDNN is a Deep Neural Network library built with cuDNN primitives specifically
The main scope is to do high performance inference on already trained models.
this branch actually work on every NVIDIA GPU that support the dependencies:
* CUDA 9
* CUDNN 7.105
* TENSORRT 4.02
* CUDA 10.0
* CUDNN 7.603
* TENSORRT 6.01
* OPENCV 4.1
## Dependencies