提交 4eb756a5 authored 作者: Frederic Bastien's avatar Frederic Bastien
上级 accb6776
......@@ -129,7 +129,8 @@ def conv2d(input, filters, image_shape=None, filter_shape=None,
if image_shape and filter_shape:
try:
assert image_shape[1] == filter_shape[1]
if image_shape[1] is not None and filter_shape[1] is not None:
assert image_shape[1] == filter_shape[1]
except Exception:
print('image ', image_shape, ' filters ', filter_shape)
raise
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论