From e9b94cabbf46578058407c2b051a8e13f55e0420 Mon Sep 17 00:00:00 2001
From: wwhu <wwhu@foxmail.com>
Date: Tue, 13 Jun 2017 19:41:25 +0800
Subject: [PATCH] fix bug

---
 image_classification/resnet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/image_classification/resnet.py b/image_classification/resnet.py
index eeed7141..ca9330e6 100644
--- a/image_classification/resnet.py
+++ b/image_classification/resnet.py
@@ -51,7 +51,7 @@ def bottleneck(input, ch_in, ch_out, stride):
 def layer_warp(block_func, input, ch_in, ch_out, count, stride):
     conv = block_func(input, ch_in, ch_out, stride)
     for i in range(1, count):
-        conv = block_func(conv, ch_in, ch_out, 1)
+        conv = block_func(conv, ch_out, ch_out, 1)
     return conv
 
 
-- 
GitLab