fix bug

e9b94cab · wwhu · 208ca38a · e9b94cab
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

image_classification/resnet.py image_classification/resnet.py +1 -1

未找到文件。
--- 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