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

avoid a warning

Browse Source
This commit is contained in:
Mike Innes 2018-10-05 13:59:58 +01:00
parent f08b6f80d2
commit 69afdd61a6
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tracker/Tracker.jl
View File

@ -108,10 +108,8 @@ param(xs::AbstractArray) = TrackedArray(float.(xs))
param(x::TrackedReal) = track(identity, x) param(x::TrackedReal) = track(identity, x)
param(x::TrackedArray) = track(identity, x) param(x::TrackedArray) = track(identity, x)
import NNlib.cudata
import Adapt.adapt import Adapt.adapt
cudata(x::TrackedArray) = data(x)
adapt(T, xs::TrackedArray) = param(adapt(T, data(xs))) adapt(T, xs::TrackedArray) = param(adapt(T, data(xs)))
end end