From 715b6f0d288937a49540db47e7cf8af698ee8266 Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Wed, 27 Nov 2019 13:52:28 +0900
Subject: [PATCH] Fix LBPCriterion #316

---
 lib/LBPCriterion.lua | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/LBPCriterion.lua b/lib/LBPCriterion.lua
index 3331fa5..8bd792b 100644
--- a/lib/LBPCriterion.lua
+++ b/lib/LBPCriterion.lua
@@ -12,8 +12,8 @@ local function create_filters(ch, n, k, layers)
       if i == 1 then
 	 -- channel identity
 	 for j = 1, ch do
-	    filter.weight[i]:fill(0)
-	    filter.weight[i][i][math.floor(k/2)+1][math.floor(k/2)+1] = 1
+	    filter.weight[j]:fill(0)
+	    filter.weight[j][j][math.floor(k/2)+1][math.floor(k/2)+1] = 1
 	 end
       end
       model:add(filter)