From f617aa68ecbbb2f27e5c6d490235f211f27184e7 Mon Sep 17 00:00:00 2001 From: Manikya Bardhan Date: Tue, 1 Mar 2022 17:20:11 +0530 Subject: [PATCH] Update src/layers/basic.jl Co-authored-by: Michael Abbott <32575566+mcabbott@users.noreply.github.com> --- src/layers/basic.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/layers/basic.jl b/src/layers/basic.jl index 9055ee5865..e3b5edde87 100644 --- a/src/layers/basic.jl +++ b/src/layers/basic.jl @@ -476,7 +476,8 @@ function Base.show(io::IO, m::Parallel) end """ - Embedding(in => out; init=randn) + Embedding(in => out; init=randn32) + Embedding(weight::AbstractMatrix) A lookup table that stores embeddings of dimension `out` for a vocabulary of size `in`.