yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-11-01 10:20:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

pep8 fixes

Browse Source
This commit is contained in:
rasbt 2024-05-13 07:50:51 -05:00
parent 13e4282567
commit c95abad6d1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ch06/01_main-chapter-code/gpt-class-finetune.py
View File

@ -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")
plot_values(epochs_tensor, examples_seen_tensor, train_accs, val_accs, label="accuracy")