X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=flatparam.py;h=57a872038408969605aa971af59d41d2ad2257c5;hp=e0627b2d7bdf0ef629153467bd6370a003a6a537;hb=291a994ff33490a614a4b137a17af444c79581fe;hpb=5db6798d929c15e1517ec10c1a9211f870ec977e

diff --git a/flatparam.py b/flatparam.py
index e0627b2..57a8720 100755
--- 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)
     )
 )