Skip to content

Commit 2f2bea7

Browse files
author
Avik Pal
committed
Fix failing tests
1 parent 5ed8a5b commit 2f2bea7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/conv.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -101,12 +101,12 @@ end
101101
@test squeeze(z[:,:,:,i:i], (4)) == squeeze(res, (3))
102102
end
103103

104-
@test size(∇depthwiseconv_filter(rand(2,2,2,1), x, w)) == size(w)
105-
@test size(∇depthwiseconv_data(rand(2,2,2,1), x, w)) == size(x)
104+
@test size(∇depthwiseconv_filter(rand(2,2,4,1), x, w)) == size(w)
105+
@test size(∇depthwiseconv_data(rand(2,2,4,1), x, w)) == size(x)
106106

107107
# Test for the stride/pad for backward pass
108108
y = depthwiseconv(x,w,stride=2,pad=1)
109-
@test size(y) == (2,2,2,1)
109+
@test size(y) == (2,2,4,1)
110110
@test size(∇depthwiseconv_filter(rand(size(y)), x, w, stride=2, pad=1)) == size(w)
111111
@test size(∇depthwiseconv_data(rand(size(y)), x, w, stride=2, pad=1)) == size(x)
112112
end

0 commit comments

Comments
 (0)