DiffEqFlux.jl icon indicating copy to clipboard operation
DiffEqFlux.jl copied to clipboard

missing show methods for neural DE layers

Open CarloLucibello opened this issue 4 years ago • 1 comments

These prints don't look very good:

julia> using DiffEqFlux, Flux

julia> NeuralODE(Dense(2,2), (0f0,1f0))
(::NeuralODE{Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}, Vector{Float32}, Flux.var"#122#123"{Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}}, Tuple{Float32, Float32}, Tuple{}, Base.Iterators.Pairs{Union{}, Union{}, Tuple{}, NamedTuple{(), Tuple{}}}}) (generic function with 2 methods)

julia> NeuralODE(Chain(Dense(2,2),Dense(2,2)), (0f0,1f0))
(::NeuralODE{Chain{Tuple{Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}, Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}}}, Vector{Float32}, Flux.var"#122#123"{Chain{Tuple{Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}, Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}}}}, Tuple{Float32, Float32}, Tuple{}, Base.Iterators.Pairs{Union{}, Union{}, Tuple{}, NamedTuple{(), Tuple{}}}}) (generic function with 2 methods)

CarloLucibello avatar Oct 16 '21 11:10 CarloLucibello

Agreed 😅 .

ChrisRackauckas avatar Oct 16 '21 12:10 ChrisRackauckas