From: Francois Fleuret Date: Thu, 10 Sep 2020 17:50:16 +0000 (+0200) Subject: Update. X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=291a994ff33490a614a4b137a17af444c79581fe Update. --- 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) ) )