Batchnorm eps fix, works on jetpack 4.3

2020-01-15 18:06:02 +01:00
parent 57d7743f7e
commit c32a0be257
5 changed files with 13 additions and 9 deletions
@@ -17,10 +17,6 @@ Network::Network(dataDim_t input_dim) {
             <<", CUDNN v"<<cu_ver<<")\n";
    dataType = CUDNN_DATA_FLOAT;
    tensorFormat = CUDNN_TENSOR_NCHW;
-
-    checkCUDNN( cudnnCreate(&cudnnHandle) );
-    checkERROR( cublasCreate(&cublasHandle) );
-
    num_layers = 0;

    fp16 = false;
@@ -38,6 +34,11 @@ Network::Network(dataDim_t input_dim) {
        std::cout<<COL_REDB<<"!! FP16 INERENCE ENABLED !!"<<COL_END<<"\n";
    if(dla)
        std::cout<<COL_GREENB<<"!! DLA INERENCE ENABLED !!"<<COL_END<<"\n";
+
+
+    checkCUDNN( cudnnCreate(&cudnnHandle) );
+    checkERROR( cublasCreate(&cublasHandle) );
+
 }

 Network::~Network() {