Skip to content

Commit

Permalink
Fix doctests
Browse files Browse the repository at this point in the history
  • Loading branch information
Saransh-cpp committed Oct 6, 2022
1 parent 289aa9d commit 95457d1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/layers/basic.jl
Original file line number Diff line number Diff line change
Expand Up @@ -711,7 +711,7 @@ Embedding(1000 => 4) # 4_000 parameters
julia> vocab_idxs = [1, 722, 53, 220, 3];
julia> x = Flux.onehotbatch(vocab_idxs, 1:vocab_size);
summary(x);
summary(x)
"1000×5 OneHotMatrix(::Vector{UInt32}) with eltype Bool"
julia> model(x) |> summary
Expand Down

0 comments on commit 95457d1

Please sign in to comment.