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)
|
back!(s::Chain, ∇) = foldr((m, ∇) -> back!(m, ∇), ∇, s.layers)
|
||||||
update!(s::Chain, η) = foreach(l -> update!(l, η), 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) =
|
graph(s::Chain) =
|
||||||
foldl((v, m) -> vertex(m, v), constant(inputnode(1)), s.layers)
|
foldl((v, m) -> vertex(m, v), constant(inputnode(1)), s.layers)
|
||||||
|
|
||||||
|
@ -11,9 +11,6 @@ end
|
|||||||
Dense(in::Integer, out::Integer; init = initn) =
|
Dense(in::Integer, out::Integer; init = initn) =
|
||||||
Dense(init(in, out), init(1, out))
|
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
|
@net type Sigmoid
|
||||||
layer::Model
|
layer::Model
|
||||||
x -> σ(layer(x))
|
x -> σ(layer(x))
|
||||||
|
@ -36,10 +36,6 @@ accumulate!(x, Δ) = x
|
|||||||
|
|
||||||
@forward Param.x Base.size
|
@forward Param.x Base.size
|
||||||
|
|
||||||
function Base.show(io::IO, p::Param)
|
|
||||||
print(io, "Param", size(p.x))
|
|
||||||
end
|
|
||||||
|
|
||||||
# Anonymous models
|
# Anonymous models
|
||||||
|
|
||||||
export Capacitor
|
export Capacitor
|
||||||
@ -51,5 +47,3 @@ end
|
|||||||
# TODO: Julia implementation that interprets the graph
|
# TODO: Julia implementation that interprets the graph
|
||||||
|
|
||||||
graph(cap::Capacitor) = cap.graph
|
graph(cap::Capacitor) = cap.graph
|
||||||
|
|
||||||
Base.show(io::IO, ::Capacitor) = print(io, "Flux.Capacitor(...)")
|
|
||||||
|
Loading…
Reference in New Issue
Block a user