From: François Fleuret Date: Fri, 14 Jun 2024 09:34:41 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;ds=sidebyside;p=picoclvr.git Update. --- 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)