Merge pull request #146 from Saransh-cpp/mnist-update

Use the latest API of MLDatasets.MNIST
FluxML · Oct 7, 2022 · 4be8e6c · 4be8e6c
2 parents e36b94a + a94257b
commit 4be8e6c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/_posts/2021-01-21-data-loader.md b/tutorials/_posts/2021-01-21-data-loader.md
@@ -34,8 +34,8 @@ using Flux: onehotbatch
 We load the MNIST train and test data from MLDatasets:
 
 ```julia
-train_x, train_y = MNIST.traindata(Float32)
-test_x, test_y = MNIST.testdata(Float32)
+train_x, train_y = MNIST(:train)[:]
+test_x, test_y = MNIST(:test)[:]
 ```
 <br>