We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f4f2b39 commit 320cad5Copy full SHA for 320cad5
src/layers/show.jl
@@ -88,7 +88,7 @@ end
88
_show_leaflike(x) = isleaf(x) # mostly follow Functors, except for:
89
_show_leaflike(::Tuple{Vararg{<:Number}}) = true # e.g. stride of Conv
90
_show_leaflike(::Tuple{Vararg{<:AbstractArray}}) = true # e.g. parameters of LSTMcell
91
-_show_leaflike(::Diagonal) = true # appears inside LayerNorm
+_show_leaflike(::Scale) = true # appears inside LayerNorm
92
93
_show_children(x) = trainable(x) # except for layers which hide their Tuple:
94
_show_children(c::Chain) = c.layers
0 commit comments