don't print reams of data
This commit is contained in:
parent
9675dc765d
commit
fd67383494
|
@ -5,6 +5,8 @@ type MXModel
|
|||
exec::mx.Executor
|
||||
end
|
||||
|
||||
Base.show(io::IO, ::MXModel) = print(io, "MXModel(...)")
|
||||
|
||||
mxdims(dims::NTuple) =
|
||||
length(dims) == 1 ? (1, dims...) : reverse(dims)
|
||||
|
||||
|
|
|
@ -9,6 +9,8 @@ end
|
|||
Dense(in::Integer, out::Integer; init = randn) =
|
||||
Dense(init(out, in), init(out))
|
||||
|
||||
Base.show(io::IO, ::Dense) = print(io, "Flux.Dense(...)")
|
||||
|
||||
@model type Sigmoid
|
||||
layer::Model
|
||||
x -> σ(layer(x))
|
||||
|
|
|
@ -50,3 +50,5 @@ back!(cap::Capacitor, args...) = cap.backward(args...)
|
|||
update!(cap::Capacitor, η) = cap.update(η)
|
||||
|
||||
graph(cap::Capacitor) = cap.graph
|
||||
|
||||
Base.show(io::IO, ::Capacitor) = print(io, "Flux.Capacitor(...)")
|
||||
|
|
Loading…
Reference in New Issue