Re:GPT-2: 6-Month Follow-Up
by newhaus1994 on Tuesday August 20, 2019 @05:43PM
(#20752526)
I'm the lead researcher on the Middlebury Institute project looking at fine-tuning the bigger models, and I got OOM on 745M and 1.5B originally. I had to get an Azure instance with 24GB VRAM to handle it (using nshepperd's codebase). It works, but takes a while (~500 epochs takes 12 hours on a 100k word training dataset).
[ Parent
| on HN ]
Ouch! So 11GB is nowhere close to being enough, then. I wonder if even switching to FP16 will be adequate?
Re:GPT-2: 6-Month Follow-Up
by newhaus1994 on Tuesday August 20, 2019 @07:09PM
(#20753177)
Might be able to get 745M down to work on a single GPU. I'm definitely not using all 24GB, so fp16 might be able to get it down enough.