From 0e74c10ba17f2f969072f3989579c0d6f47f1cbb Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 14 Mar 2023 21:05:11 +0100 Subject: [PATCH] Update --- beaver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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): -- 2.20.1