diff --git a/docs/src/gpu.md b/docs/src/gpu.md index 1ef7a827..63891e1d 100644 --- a/docs/src/gpu.md +++ b/docs/src/gpu.md @@ -32,4 +32,4 @@ m = mapleaves(cu, m) d(cu(rand(10))) ``` -The [mnist example](https://github.com/FluxML/model-zoo/blob/master/mnist/mnist.jl) contains the code needed to run the model on the GPU; just uncomment the lines after `using CuArrays`. +The [mnist example](https://github.com/FluxML/model-zoo/blob/master/mnist/mlp.jl) contains the code needed to run the model on the GPU; just uncomment the lines after `using CuArrays`. diff --git a/src/Flux.jl b/src/Flux.jl index 179d09c2..522f83c5 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -18,6 +18,7 @@ export σ, sigmoid, relu, leakyrelu, elu, swish, softmax, logsoftmax, include("tracker/Tracker.jl") using .Tracker +import .Tracker: data, value include("optimise/Optimise.jl") using .Optimise diff --git a/src/tracker/Tracker.jl b/src/tracker/Tracker.jl index aa2bc6ea..5686c7a2 100644 --- a/src/tracker/Tracker.jl +++ b/src/tracker/Tracker.jl @@ -65,7 +65,7 @@ value(x::TrackedScalar) = data(x)[] Base.:(==)(x::TrackedArray, y) = value(x) == y Base.:(==)(y, x::TrackedArray) = y == value(x) -Base.:(==)(x::TrackedArray, y::TrackedArray) = value(x) == value(x) +Base.:(==)(x::TrackedArray, y::TrackedArray) = value(x) == value(y) Base.isless(x::TrackedScalar, y) = isless(value(x), y) Base.isless(x, y::TrackedScalar) = isless(x, value(y))