remove custom show
This commit is contained in:
parent
dab491955f
commit
e4a6ca5f9e
|
@ -26,12 +26,6 @@ end
|
|||
back!(s::Chain, ∇) = foldr((m, ∇) -> back!(m, ∇), ∇, s.layers)
|
||||
update!(s::Chain, η) = foreach(l -> update!(l, η), s.layers)
|
||||
|
||||
function Base.show(io::IO, c::Chain)
|
||||
print(io, "Chain(")
|
||||
print_joined(io, c.layers, ", ")
|
||||
print(io, ")")
|
||||
end
|
||||
|
||||
graph(s::Chain) =
|
||||
foldl((v, m) -> vertex(m, v), constant(inputnode(1)), s.layers)
|
||||
|
||||
|
|
|
@ -11,9 +11,6 @@ end
|
|||
Dense(in::Integer, out::Integer; init = initn) =
|
||||
Dense(init(in, out), init(1, out))
|
||||
|
||||
Base.show(io::IO, d::Dense) =
|
||||
print(io, "Dense($(size(d.W.x,1)),$(size(d.W.x,2)))")
|
||||
|
||||
@net type Sigmoid
|
||||
layer::Model
|
||||
x -> σ(layer(x))
|
||||
|
|
|
@ -36,10 +36,6 @@ accumulate!(x, Δ) = x
|
|||
|
||||
@forward Param.x Base.size
|
||||
|
||||
function Base.show(io::IO, p::Param)
|
||||
print(io, "Param", size(p.x))
|
||||
end
|
||||
|
||||
# Anonymous models
|
||||
|
||||
export Capacitor
|
||||
|
@ -51,5 +47,3 @@ end
|
|||
# TODO: Julia implementation that interprets the graph
|
||||
|
||||
graph(cap::Capacitor) = cap.graph
|
||||
|
||||
Base.show(io::IO, ::Capacitor) = print(io, "Flux.Capacitor(...)")
|
||||
|
|
Loading…
Reference in New Issue