From 22415499c0a91922e51f9e2cade009fd404351dc Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Fri, 14 Jun 2024 11:34:41 +0200 Subject: [PATCH] Update. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 37515b5..3ff64b7 100755 --- a/main.py +++ b/main.py @@ -844,7 +844,7 @@ for n_epoch in range(nb_epochs_finished, args.nb_epochs): input = input.to(device) bs = model(mygpt.BracketedSequence(input)) - output_ar = bs.x + output = bs.x loss = F.cross_entropy(output.transpose(1, 2), input) -- 2.20.1