From 4942a6d07bd300f8218e928905a42f95e8c8a690 Mon Sep 17 00:00:00 2001
From: lltcggie <lltcggie@gmail.com>
Date: Sun, 7 Jun 2015 04:52:35 +0900
Subject: [PATCH] =?UTF-8?q?batch=5Fsize=E3=82=92=E6=8C=87=E5=AE=9A?=
 =?UTF-8?q?=E3=81=97=E3=81=9F=E5=A0=B4=E5=90=88=E3=80=81=E3=83=8D=E3=83=83?=
 =?UTF-8?q?=E3=83=88=E3=81=AB=E5=85=A5=E5=8A=9B=E3=81=99=E3=82=8B=E3=83=87?=
 =?UTF-8?q?=E3=83=BC=E3=82=BF=E3=81=AE=E3=82=B5=E3=82=A4=E3=82=BA=E3=81=8C?=
 =?UTF-8?q?=E9=96=93=E9=81=95=E3=81=A3=E3=81=A6=E3=81=84=E3=81=9F=E3=81=AE?=
 =?UTF-8?q?=E3=82=92=E5=84=AA=E5=8B=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 common/waifu2x.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/common/waifu2x.cpp b/common/waifu2x.cpp
index 2a243e9..eeddb5b 100644
--- a/common/waifu2x.cpp
+++ b/common/waifu2x.cpp
@@ -455,7 +455,7 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructImage(boost::shared_ptr<caffe::Net<fl
 			}
 
 			// ネットワークに画像を入力
-			input_layer->Reset(block, dummy_data, input_block_plane_size * batch_size);
+			input_layer->Reset(block, dummy_data, input_block_plane_size * processNum);
 
 			// 計算
 			auto out = net->ForwardPrefilled(nullptr);