Update.

diff --git a/flatparam.py b/flatparam.py
index e0627b2..57a8720 100755 (executable)
--- a/flatparam.py
+++ b/flatparam.py
@@ -34,7 +34,8 @@ model = nn.Sequential(
     nn.ReLU(),
     nn.Sequential(
         nn.Linear(4, 4),
-        nn.ReLU(), nn.Linear(4, 2)
+        nn.ReLU(),
+        nn.Linear(4, 2)
     )
 )