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

View File

@ -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`.