Skip to content

Commit bfc5bb0

Browse files
committed
rebase
1 parent f263f0c commit bfc5bb0

File tree

1 file changed

+0
-19
lines changed

1 file changed

+0
-19
lines changed

test/layers/conv.jl

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -101,23 +101,4 @@ end
101101
randn(Float32, 10,10,1,1) |> Conv((6,1), 1=>1, Flux.σ)
102102
true
103103
end
104-
@testset "CrossCor" begin
105-
x = rand(Float32, 28, 28, 1, 1)
106-
w = rand(2,2,1,1)
107-
y = CrossCor(w, [0.0])
108-
x_pred = y(x)
109104

110-
@test sum(w .* x[1:2, 1:2, :, :]) == x_pred[1, 1, 1, 1]
111-
112-
r = zeros(Float32, 28, 28, 1, 5)
113-
m = Chain(
114-
CrossCor((2, 2), 1=>16, relu),
115-
MaxPool((2,2)),
116-
CrossCor((2, 2), 16=>8, relu),
117-
MaxPool((2,2)),
118-
x -> reshape(x, :, size(x, 4)),
119-
Dense(288, 10), softmax)
120-
121-
@test size(m(r)) == (10, 5)
122-
123-
end

0 commit comments

Comments
 (0)