From 291a994ff33490a614a4b137a17af444c79581fe Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Thu, 10 Sep 2020 19:50:16 +0200 Subject: [PATCH] Update. --- flatparam.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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) ) ) -- 2.39.5