fix mnist example
This commit is contained in:
parent
dc3485972a
commit
574fccd44d
@ -1,5 +1,5 @@
|
|||||||
using Flux, MNIST
|
using Flux, MNIST
|
||||||
using Flux: accuracy, onehot
|
using Flux: accuracy, onehot, tobatch
|
||||||
|
|
||||||
data = [(trainfeatures(i), onehot(trainlabel(i), 0:9)) for i = 1:60_000]
|
data = [(trainfeatures(i), onehot(trainlabel(i), 0:9)) for i = 1:60_000]
|
||||||
train = data[1:50_000]
|
train = data[1:50_000]
|
||||||
|
Loading…
Reference in New Issue
Block a user