提交 bea725bb 编写于 作者: H huangjun12 提交者: chajchaj

fix lrn bug in export model

上级 4d198acb
......@@ -489,7 +489,7 @@ def local_response_norm(x,
format(dim))
for i, sz in enumerate(sizes):
if not sz > 0:
if not sz > 0 and i > 0:
raise ValueError("Expected every dim's size to be larger than 0, "
"but the size of the {}-th dim is {}".format(i,
sz))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册