This commit is contained in:
Mike Innes 2018-08-28 10:54:50 +01:00
parent 89bca2d98d
commit 2ca189bc96
4 changed files with 4 additions and 4 deletions

View File

@ -37,4 +37,4 @@ include("data/Data.jl")
@init @require CuArrays="3a865a2d-5b23-5a0f-bc46-62713ec82fae" include("cuda/cuda.jl")
end # module
end # module

View File

@ -20,4 +20,4 @@ using Flux.Tracker: TrackedArray
Param(x::TrackedArray) = Param(x.data, x.grad)
# Base.convert(::Type{Param}, x::TrackedArray) = Param(x.data, x.grad)
end
end

View File

@ -93,4 +93,4 @@ macro epochs(n, ex)
@info "Epoch $i"
$(esc(ex))
end)
end
end

View File

@ -152,4 +152,4 @@ function jacobian(m,x)
xp.grad .*= 0 # Reset gradient accumulator
end
J'
end
end