yolo3 86 route error

2018-12-23 16:20:17 +01:00
parent 3bd725801d
commit 2e8d0b1002
4 changed files with 29 additions and 24 deletions
@@ -9,6 +9,11 @@ Shortcut::Shortcut(Network *net, Layer *backLayer) : Layer(net) {

    this->backLayer = backLayer;
    checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );
+
+    if( backLayer->output_dim.c != input_dim.c ||
+        backLayer->output_dim.w != input_dim.w ||
+        backLayer->output_dim.h != input_dim.h   )
+        FatalError("Shortcut dim missmatch");
 }

 Shortcut::~Shortcut() {