From: François Fleuret <francois@fleuret.org>
Date: Tue, 14 Mar 2023 20:05:11 +0000 (+0100)
Subject: Update
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=beaver.git;a=commitdiff_plain;h=0e74c10ba17f2f969072f3989579c0d6f47f1cbb

Update
---

diff --git a/beaver.py b/beaver.py
index 1408f0b..54510f0 100755
--- a/beaver.py
+++ b/beaver.py
@@ -175,7 +175,7 @@ def one_shot(gpt, task):
     model = nn.Sequential(
         nn.Linear(args.dim_model, args.dim_model),
         nn.ReLU(),
-        nn.Linear(args.dim_model, 4)
+        nn.Linear(args.dim_model, 4),
     ).to(device)
 
     for n_epoch in range(args.nb_epochs):