未验证 提交 815f20b5 编写于 作者: L littletomatodonkey 提交者: GitHub

Merge pull request #834 from tink2123/fix_srn

Fix srn
...@@ -100,11 +100,10 @@ class ResNet(): ...@@ -100,11 +100,10 @@ class ResNet():
conv = self.basic_block( conv = self.basic_block(
input=conv, input=conv,
num_filters=num_filters[block], num_filters=num_filters[block],
stride=stride, stride=stride_list[block] if i == 0 else 1,
if_first=block == i == 0, is_first=block == i == 0,
name=conv_name) name=conv_name)
F.append(conv) F.append(conv)
base = F[-1] base = F[-1]
for i in [-2, -3]: for i in [-2, -3]:
b, c, w, h = F[i].shape b, c, w, h = F[i].shape
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册