AI-hardware en softwarestack
Een model met 7 miljard (7B) parameters wordt getraind waarbij gewichten, gradiënten en de twee toestanden van Adam allemaal in FP32 (4 bytes) worden bijgehouden. Welke waarde benadert het benodigde geheugen, exclusief activaties, het best?
- ACirca 28 GB
- BCirca 112 GB
- CCirca 14 GB
Antwoord en toelichting tonen
Antwoord: B. Circa 112 GB
Per parameter is dat 4 bytes voor het gewicht, 4 bytes voor de gradiënt en 4 bytes elk voor het eerste en tweede voortschrijdend gemiddelde van Adam, samen 16 bytes. 7 miljard × 16 bytes = 112 miljard bytes, circa 112 GB, waarbij nog het geheugen voor activaties komt. Omdat dit vaak niet in het geheugen van één GPU past, worden maatregelen gebruikt zoals het verdelen van de optimizertoestand over meerdere GPU's en het bijhouden met lagere precisie.