aparler/c-llm
1
0
Fork 0
mirror of https://github.com/karpathy/llm.c.git synced 2026-07-27 20:25:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix argparse comment too

Browse source
This commit is contained in:
Andrej Karpathy 2024-05-31 22:06:48 +00:00
parent 60a1037db9
commit 1de3bef0ac
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
train_gpt2.cu
View file

@ -3024,7 +3024,7 @@ void error_usage() {
fprintf(stderr, " -w <int> keep f32 copy of weights for the optimizer? (default: 1)\n");
// memory management
fprintf(stderr, " -z <int> zero_stage, Zero Optimization Stage, 0,1,2,3 (default = 0)\n");
fprintf(stderr, " -r <int> recompute: saves memory at cost of speed. (default = 1), 0 = none. 1 = recompute gelu\n");
fprintf(stderr, " -r <int> recompute: less memory but less speed. (default = 1), 0|1|2 = none,gelu,gelu+ln\n");
exit(EXIT_FAILURE);
}