NetworkRT (deallocations to be done)

2017-08-03 12:16:57 +02:00
parent e8355cee67
commit 4526e2767a
11 changed files with 344 additions and 34 deletions
@@ -15,6 +15,8 @@ Conv2d::Conv2d( Network *net, int out_ch, int kernelH, int kernelW,
    this->kernelW = kernelW;
    this->strideH = strideH;
    this->strideW = strideW;
+    this->paddingH = paddingH;
+    this->paddingW = paddingW;

    checkCUDNN( cudnnCreateFilterDescriptor(&filterDesc) );
    checkCUDNN( cudnnCreateConvolutionDescriptor(&convDesc) );