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

link fix

Browse Source
This commit is contained in:
Mike J Innes 2018-02-01 16:01:48 +00:00
parent 106502a75d
commit 62c3376feb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/src/gpu.md
View File

@ -32,4 +32,4 @@ m = mapleaves(cu, m)
d(cu(rand(10))) 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`.