TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

don't print reams of data

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/backend/mxnet/model.jl
View File

@ -5,6 +5,8 @@ type MXModel
exec::mx.Executor exec::mx.Executor
end end
Base.show(io::IO, ::MXModel) = print(io, "MXModel(...)")
mxdims(dims::NTuple) = mxdims(dims::NTuple) =
length(dims) == 1 ? (1, dims...) : reverse(dims) length(dims) == 1 ? (1, dims...) : reverse(dims)

2
src/layers/dense.jl
View File

@ -9,6 +9,8 @@ end
Dense(in::Integer, out::Integer; init = randn) = Dense(in::Integer, out::Integer; init = randn) =
Dense(init(out, in), init(out)) Dense(init(out, in), init(out))
Base.show(io::IO, ::Dense) = print(io, "Flux.Dense(...)")
@model type Sigmoid @model type Sigmoid
layer::Model layer::Model
x -> σ(layer(x)) x -> σ(layer(x))

2
src/model.jl
View File

@ -50,3 +50,5 @@ back!(cap::Capacitor, args...) = cap.backward(args...)
update!(cap::Capacitor, η) = cap.update(η) update!(cap::Capacitor, η) = cap.update(η)
graph(cap::Capacitor) = cap.graph graph(cap::Capacitor) = cap.graph
Base.show(io::IO, ::Capacitor) = print(io, "Flux.Capacitor(...)")