inf loss at big batch
- 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