small fix in mnist classifier

lecopivo · Dec 5, 2023 · 51ac69e · 51ac69e
1 parent a9f867b
commit 51ac69e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/SciLean/Modules/ML/MNIST.lean b/SciLean/Modules/ML/MNIST.lean
@@ -17,9 +17,9 @@ def mnist (w x) :=
       |> map gelu
       |> avgPool
       |> dense 100 w₂ b₂
-      |> map gelu
-      |> dense 10 w₃ b₃) w x
-      |> softMax 1
+      |> map gelu 
+      |> dense 10 w₃ b₃
+      |> softMax 1) w x
 
 #generate_revDeriv mnist w x
   prop_by unfold mnist; simp[mnist.match_1]; fprop