fix

2019-10-30 09:42:54 +01:00
parent 33f36ab204
commit d6d93a74f8
1 changed files with 2 additions and 2 deletions
@@ -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;