@@ -68,15 +68,15 @@ function _big_finale(io::IO, m)
68
68
noncnt = _childarray_sum (_-> 1 , m) - length (ps)
69
69
if noncnt > 0
70
70
nonparam = underscorise (_childarray_sum (length, m) - sum (length, ps))
71
- printstyled (io, " " ^ 19 , " # Total: " , length (ps), " trainable arrays, " , pars, " parameters,\n " ; color= :light_black )
72
- printstyled (io, " " ^ 20 , " # plus " , noncnt, " non-trainable, " , nonparam, " parameters, total size " , bytes; color= :light_black )
71
+ printstyled (io, " " ^ 09 , " # Total: " , length (ps), " trainable arrays, with " , pars, " parameters,\n " ; color= :light_black )
72
+ printstyled (io, " " ^ 10 , " # plus " , noncnt, " non-trainable, " , nonparam, " parameters, total size " , bytes; color= :light_black )
73
73
else
74
74
printstyled (io, " " ^ 19 , " # Total: " , length (ps), " arrays, " , pars, " parameters, " , bytes; color= :light_black )
75
75
end
76
76
end
77
77
end
78
78
79
- _childarray_sum (f, x:: AbstractArray ) = f (x) # count includes non-trainable arrays excluded from params
79
+ _childarray_sum (f, x:: AbstractArray ) = f (x)
80
80
_childarray_sum (f, x) = isleaf (x) ? 0 : sum (y -> _childarray_sum (f, y), Functors. children (x))
81
81
82
82
# utility functions
0 commit comments