Skip to content

Commit 3e9c481

Browse files
committed
note non-trainables on every layer which has such
1 parent 9690dfb commit 3e9c481

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/layers/show.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,10 @@ function _layer_show(io::IO, layer, indent::Int=0)
5555
if !isempty(params(layer))
5656
print(io, " "^max(2, (indent==0 ? 20 : 39) - indent - length(str)))
5757
printstyled(io, "# ", underscorise(sum(length, params(layer))), " parameters"; color=:light_black)
58+
nonparam = _childarray_sum(length, layer) - sum(length, params(layer))
59+
if nonparam > 0
60+
printstyled(io, ", plus ", underscorise(nonparam); color=:light_black)
61+
end
5862
_nan_show(io, params(layer))
5963
end
6064
indent==0 || println(io)

0 commit comments

Comments
 (0)