From d11f22a98f60a2735a78d81e540823f3731554dd Mon Sep 17 00:00:00 2001 From: lltcggie Date: Mon, 4 Jul 2016 01:16:23 +0900 Subject: [PATCH] =?UTF-8?q?batch=5Fsize=E3=82=922=E4=BB=A5=E4=B8=8A?= =?UTF-8?q?=E3=81=AB=E3=81=99=E3=82=8B=E3=81=A8=E5=BC=B7=E5=88=B6=E7=B5=82?= =?UTF-8?q?=E4=BA=86=E3=81=99=E3=82=8B=E3=83=90=E3=82=B0=E3=82=92=E4=BF=AE?= =?UTF-8?q?=E6=AD=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- common/cNet.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/common/cNet.cpp b/common/cNet.cpp index b6e5c0a..8819c53 100644 --- a/common/cNet.cpp +++ b/common/cNet.cpp @@ -577,7 +577,7 @@ int cNet::GetInputMemorySize(const int crop_w, const int crop_h, const int outer const int input_block_plane_size = input_block_width * input_block_height * mInputPlane; - return input_block_plane_size * sizeof(float); + return input_block_plane_size * batch_size * sizeof(float); } @@ -592,7 +592,7 @@ int cNet::GetOutputMemorySize(const int crop_w, const int crop_h, const int oute const int output_block_plane_size = output_block_width * output_block_height * mInputPlane; - return output_block_plane_size * sizeof(float); + return output_block_plane_size * batch_size * sizeof(float); } // ネットワークを使って画像を再構築する