DiffEqFlux.jl
DiffEqFlux.jl copied to clipboard
missing show methods for neural DE layers
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)
Agreed 😅 .