Update example/MNIST_cnn.jl

Co-authored-by: tmigot <[email protected]>
JuliaSmoothOptimizers · Nov 8, 2023 · 68ac783 · 68ac783
1 parent 71b1a0e
commit 68ac783
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/MNIST_cnn.jl b/example/MNIST_cnn.jl
@@ -57,7 +57,7 @@ model =
     Flux.flatten,
     Dense(256 => 120, relu),
     Dense(120 => 84, relu),
-    Dense(84 => 10),#TODO add softmax
+    Dense(84 => 10),
   ) |> device
 
 nlp = FluxNLPModel(model, train_loader, test_loader; loss_f = loss)