From 15fa6a84f6605af5d6f299668d416fcb20754b11 Mon Sep 17 00:00:00 2001 From: Daniel Kleine <53251018+d-kleine@users.noreply.github.com> Date: Fri, 20 Jun 2025 01:55:43 +0200 Subject: [PATCH] fixed plot_losses (#677) --- ch05/03_bonus_pretraining_on_gutenberg/pretraining_simple.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ch05/03_bonus_pretraining_on_gutenberg/pretraining_simple.py b/ch05/03_bonus_pretraining_on_gutenberg/pretraining_simple.py index d988fd4..92237a1 100644 --- a/ch05/03_bonus_pretraining_on_gutenberg/pretraining_simple.py +++ b/ch05/03_bonus_pretraining_on_gutenberg/pretraining_simple.py @@ -228,7 +228,7 @@ if __name__ == "__main__": ) epochs_tensor = torch.linspace(0, args.n_epochs, len(train_losses)) - plot_losses(epochs_tensor, tokens_seen, train_losses, val_losses, output_dir) + plot_losses(epochs_tensor, tokens_seen, train_losses, val_losses) torch.save(model.state_dict(), output_dir / "model_pg_final.pth") print(f"Maximum GPU memory allocated: {torch.cuda.max_memory_allocated() / 1e9:.2f} GB")