diff --git a/ch06/01_main-chapter-code/gpt-class-finetune.py b/ch06/01_main-chapter-code/gpt-class-finetune.py index 397b9eb..b1c7053 100644 --- a/ch06/01_main-chapter-code/gpt-class-finetune.py +++ b/ch06/01_main-chapter-code/gpt-class-finetune.py @@ -226,7 +226,7 @@ def plot_values(epochs_seen, examples_seen, train_values, val_values, label="los fig.tight_layout() # Adjust layout to make room plt.savefig(f"{label}-plot.pdf") - #plt.show() + # plt.show() if __name__ == "__main__": @@ -415,4 +415,4 @@ if __name__ == "__main__": # accuracy plot epochs_tensor = torch.linspace(0, num_epochs, len(train_accs)) examples_seen_tensor = torch.linspace(0, examples_seen, len(train_accs)) - plot_values(epochs_tensor, examples_seen_tensor, train_accs, val_accs, label="accuracy") \ No newline at end of file + plot_values(epochs_tensor, examples_seen_tensor, train_accs, val_accs, label="accuracy")