don't print reams of data

This commit is contained in:
Mike J Innes 2016-08-25 17:24:39 +01:00
parent 9675dc765d
commit fd67383494
3 changed files with 6 additions and 0 deletions

View File

@ -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)

View File

@ -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))

View File

@ -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(...)")