Skip to content

Commit 4a2ebdc

Browse files
authored
Update test-Conv.py
1 parent da9885d commit 4a2ebdc

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test-Conv.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ def main():
2121
for dimension, NativeModule in zip(DIMENSIONS, NATIVE_MODULES):
2222
for batches in range(BATCHES_START, BATCHES_END):
2323
for groups in range(GROUPS_START, GROUPS_END):
24-
for in_channels in range(IN_CHANNELS_START * groups, IN_CHANNELS_END * groups):
25-
for out_channels in range(OUT_CHANNELS_START * groups, OUT_CHANNELS_END * groups):
24+
for in_channels in range(IN_CHANNELS_START * groups, IN_CHANNELS_END * groups, groups):
25+
for out_channels in range(OUT_CHANNELS_START * groups, OUT_CHANNELS_END * groups, groups):
2626
for in_volume in numpy.ndindex(*numpy.full((dimension,), IN_VOLUME_END - IN_VOLUME_START, dtype=int)):
2727
in_volume = numpy.add(in_volume, IN_VOLUME_START)
2828
for kernel_size in numpy.ndindex(*(in_volume - 1)):

0 commit comments

Comments
 (0)