karpathy / karpathy/llm.c

inf loss at big batch

Open
#263 1 comment 0 reactions 0 assignees View on GitHub
bug
Dominant language
Cuda
Stars
31k
Forks
3.8k
PR merge metrics
No merged PRs in 30d

Description

just creating a todo. large batch sizes work now having fixed the `size_t` bug:

```
./train_gpt2cu -b 36 -v 200 -s 200 -i data/TinyStories
```

works, but 48 should fit but doesn't work

```
./train_gpt2cu -b 48 -v 200 -s 200 -i data/TinyStories
```

val loss is -nan and train loss stays at inf.

todo track down why and how to prevent

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reproducing the issue with the train_gpt2cu commands and the data/TinyStories dataset, comparing batch sizes 36 and 48. Track the source of the inf training loss and -nan validation loss; done means batch size 48 trains with finite losses and a preventative fix is identified.

Written by the indexing model from the issue text.

Assessment

Tech stack
c
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.