From a4d28107f409699ce0a10fb1cd560dfae4625bfa Mon Sep 17 00:00:00 2001 From: nihuini Date: Fri, 1 Dec 2017 11:16:01 +0800 Subject: [PATCH] check clone empty --- src/layer.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/layer.cpp b/src/layer.cpp index 36b1c4c98..a0666a586 100644 --- a/src/layer.cpp +++ b/src/layer.cpp @@ -54,6 +54,8 @@ int Layer::forward(const std::vector& bottom_blobs, std::vector& top_b for (int i = 0; i < (int)top_blobs.size(); i++) { top_blobs[i] = bottom_blobs[i].clone(); + if (top_blobs[i].empty()) + return -100; } return forward_inplace(top_blobs); @@ -65,6 +67,8 @@ int Layer::forward(const Mat& bottom_blob, Mat& top_blob) const return -1; top_blob = bottom_blob.clone(); + if (top_blob.empty()) + return -100; return forward_inplace(top_blob); }