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

newlines

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Flux.jl
View File

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

2
src/optimise/Optimise.jl
View File

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

2
src/optimise/train.jl
View File

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

2
src/utils.jl
View File

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