Skip to content

Commit 320cad5

Browse files
committed
bug fix
1 parent f4f2b39 commit 320cad5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/layers/show.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ end
8888
_show_leaflike(x) = isleaf(x) # mostly follow Functors, except for:
8989
_show_leaflike(::Tuple{Vararg{<:Number}}) = true # e.g. stride of Conv
9090
_show_leaflike(::Tuple{Vararg{<:AbstractArray}}) = true # e.g. parameters of LSTMcell
91-
_show_leaflike(::Diagonal) = true # appears inside LayerNorm
91+
_show_leaflike(::Scale) = true # appears inside LayerNorm
9292

9393
_show_children(x) = trainable(x) # except for layers which hide their Tuple:
9494
_show_children(c::Chain) = c.layers

0 commit comments

Comments
 (0)