From d6d93a74f8adfe482dbe49fad0b2772e8754639b Mon Sep 17 00:00:00 2001
From: fbagni <gattinomicino>
Date: Wed, 30 Oct 2019 09:42:54 +0100
Subject: [PATCH] fix

---
 src/NetworkRT.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/NetworkRT.cpp b/src/NetworkRT.cpp
index ce29f01..0f38f87 100644
--- a/src/NetworkRT.cpp
+++ b/src/NetworkRT.cpp
@@ -236,14 +236,14 @@ ILayer* NetworkRT::convert_layer(ITensor *input, Conv2d *l) {
     if(!l->deConv) {
         IConvolutionLayer *lRTconv = networkRT->addConvolution(*input, 
             l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b);
-        checkNULL(lRT);
+        checkNULL(lRTconv);
         lRTconv->setStride(DimsHW{l->strideH, l->strideW});
         lRTconv->setPadding(DimsHW{l->paddingH, l->paddingW});
         lRT = (ILayer*) lRTconv;
     } else {
         IDeconvolutionLayer *lRTconv = networkRT->addDeconvolution(*input, 
             l->outputs, DimsHW{l->kernelH, l->kernelW}, w, b);
-        checkNULL(lRT);
+        checkNULL(lRTconv);
         lRTconv->setStride(DimsHW{l->strideH, l->strideW});
         lRTconv->setPadding(DimsHW{l->paddingH, l->paddingW});
         lRT = (ILayer*) lRTconv;