Add computation of #parameters, #MACC, and max feature map size in the tests

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-11-23 13:07:54 +01:00
parent 04b4a69107
commit a17e7800b9
7 changed files with 58 additions and 1 deletions
@@ -73,6 +73,12 @@ DeformConv2d::DeformConv2d( Network *net, int out_ch, int deformable_group, int
    
    output_dim.c = out_ch;
    initCUDNN();
+
+    if(this->deformableGroup != 1)
+        MACC = kernelH*kernelW*output_dim.c*output_dim.w*output_dim.h;
+    else
+        MACC = input_dim.c*kernelH*kernelW*output_dim.c*output_dim.w*output_dim.h;
+        
    //allocate data for infer result
    checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );
 }