提交 4a3d8c7d authored 作者: Virgile Andreani's avatar Virgile Andreani 提交者: Ricardo Vieira

Make non-strict zip strict in test_abstract_conv

上级 fdeabc18
...@@ -1745,7 +1745,7 @@ class TestConv2dGrads: ...@@ -1745,7 +1745,7 @@ class TestConv2dGrads:
self.random_stream = np.random.default_rng(utt.fetch_seed()) self.random_stream = np.random.default_rng(utt.fetch_seed())
self.inputs_shapes = [(8, 1, 12, 12), (1, 1, 5, 5), (1, 1, 5, 6), (1, 1, 6, 6)] self.inputs_shapes = [(8, 1, 12, 12), (1, 1, 5, 5), (1, 1, 5, 6), (1, 1, 6, 6)]
self.filters_shapes = [(5, 1, 2, 2), (1, 1, 3, 3)] self.filters_shapes = [(5, 1, 2, 2), (1, 1, 3, 3)] * 2
self.subsamples = [(1, 1), (2, 2)] self.subsamples = [(1, 1), (2, 2)]
self.border_modes = ["valid", "full"] self.border_modes = ["valid", "full"]
...@@ -1764,7 +1764,7 @@ class TestConv2dGrads: ...@@ -1764,7 +1764,7 @@ class TestConv2dGrads:
# results are the same. # results are the same.
for in_shape, fltr_shape in zip( for in_shape, fltr_shape in zip(
self.inputs_shapes, self.filters_shapes, strict=False self.inputs_shapes, self.filters_shapes, strict=True
): ):
for bm in self.border_modes: for bm in self.border_modes:
for ss in self.subsamples: for ss in self.subsamples:
...@@ -1828,7 +1828,7 @@ class TestConv2dGrads: ...@@ -1828,7 +1828,7 @@ class TestConv2dGrads:
# results are the same. # results are the same.
for in_shape, fltr_shape in zip( for in_shape, fltr_shape in zip(
self.inputs_shapes, self.filters_shapes, strict=False self.inputs_shapes, self.filters_shapes, strict=True
): ):
for bm in self.border_modes: for bm in self.border_modes:
for ss in self.subsamples: for ss in self.subsamples:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论