Skip to content

Commit a86185a

Browse files
fix
1 parent 9ce7347 commit a86185a

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/layers/basic.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -398,8 +398,8 @@ If called with multiple inputs, they are `zip`ped with the layers, thus `Paralle
398398
399399
```jldoctest
400400
julia> model = Chain(Dense(3, 5),
401-
Parallel(vcat, Dense(5, 4), Chain(Dense(5, 7), Dense(7, 4))),
402-
Dense(8, 17));
401+
Parallel(vcat, Dense(5, 4), Chain(Dense(5, 7), Dense(7, 4))),
402+
Dense(8, 17));
403403
404404
julia> size(model(rand(3)))
405405
(17,)

test/cuda/layers.jl

+1
Original file line numberDiff line numberDiff line change
@@ -271,4 +271,5 @@ end
271271
gs = gradient(() -> sum(tanh.(m(x))), params(m))
272272
gs_g = gradient(() -> sum(tanh.(m_g(x_g))), params(m_g))
273273
@test collect(gs_g[m_g.weight]) gs[m.weight]
274+
end
274275
end

0 commit comments

Comments
 (0)