tracker submodule
This commit is contained in:
parent
15c4f38130
commit
1889ccd316
@ -19,8 +19,10 @@ export @net, unroll, unroll1, @shapes,
|
|||||||
include("core.jl")
|
include("core.jl")
|
||||||
import .FluxCore: graph
|
import .FluxCore: graph
|
||||||
|
|
||||||
|
include("Tracker/Tracker.jl")
|
||||||
|
using .Tracker
|
||||||
|
|
||||||
include("utils.jl")
|
include("utils.jl")
|
||||||
include("grad/track.jl")
|
|
||||||
include("params.jl")
|
include("params.jl")
|
||||||
|
|
||||||
include("compiler/code.jl")
|
include("compiler/code.jl")
|
||||||
|
@ -1,3 +1,5 @@
|
|||||||
|
module Tracker
|
||||||
|
|
||||||
data(x) = x
|
data(x) = x
|
||||||
|
|
||||||
struct Call{F,As<:Tuple}
|
struct Call{F,As<:Tuple}
|
||||||
@ -54,3 +56,5 @@ function Base.showarray(io::IO, X::Var, repr::Bool = true; header = true)
|
|||||||
Base.showarray(io, data(X), false, header = false)
|
Base.showarray(io, data(X), false, header = false)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
end
|
Loading…
Reference in New Issue
Block a user