tracker submodule
This commit is contained in:
parent
15c4f38130
commit
1889ccd316
@ -19,8 +19,10 @@ export @net, unroll, unroll1, @shapes,
|
||||
include("core.jl")
|
||||
import .FluxCore: graph
|
||||
|
||||
include("Tracker/Tracker.jl")
|
||||
using .Tracker
|
||||
|
||||
include("utils.jl")
|
||||
include("grad/track.jl")
|
||||
include("params.jl")
|
||||
|
||||
include("compiler/code.jl")
|
||||
|
@ -1,3 +1,5 @@
|
||||
module Tracker
|
||||
|
||||
data(x) = x
|
||||
|
||||
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)
|
||||
end
|
||||
end
|
||||
|
||||
end
|
Loading…
Reference in New Issue
Block a user