books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

User argpars utils to show default args on command line

Browse Source
This commit is contained in:
rasbt
2026-03-01 20:15:21 -06:00
parent c079904491
commit 4612d20fa8
27 changed files with 30 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ch04/08_deltanet/plot_memory_estimates_gated_deltanet.py
View File

@@ -36,7 +36,7 @@ def convert_to_gb(x):
def main():
p = argparse.ArgumentParser(description="Memory vs. Context Length: MHA vs. DeltaNet (3:1 mix)")
p = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, description="Memory vs. Context Length: MHA vs. DeltaNet (3:1 mix)")
p.add_argument("--batch", type=int, default=1)
p.add_argument("--emb_dim", type=int, default=2048)
p.add_argument("--n_heads", type=int, default=16)