Как пользоваться
- Введите параметры и токены обучения в миллиардах (2 трлн токенов = 2000).
- Введите число GPU, их пиковые TFLOP/s и ожидаемый MFU.
- Смотрите общий объём вычислений, время и GPU-часы; сравните число токенов на параметр с ориентиром Chinchilla.
Формулы
- Вычисления обучения C = 6 × N × D FLOPs
- Реальная скорость одного GPU = пик × MFU
- GPU-часы = C ÷ (пик × MFU) ÷ 3600; время по часам = GPU-часы ÷ число GPU
- Петафлоп/с-дни = C ÷ (10¹⁵ × 86 400)
- Оптимум токенов по Chinchilla ≈ 20 × N; инференс ≈ 2N FLOPs на сгенерированный токен
Пример
Модель на 7 млрд параметров, 2 трлн токенов, 512 GPU по 312 TFLOP/s при MFU 40 %:
- C = 6 × 7 × 10⁹ × 2 × 10¹² = 8,4E22 FLOPs (972,2 петафлоп/с-дня);
- каждый GPU даёт 312 × 0,4 = 124,8 TFLOP/s, поэтому нужно 8,4 × 10²² ÷ 1,248 × 10¹⁴ ÷ 3600 = 186 966 GPU-часов;
- на 512 GPU это 365,2 часа, около 15,22 дня;
- 2000 ÷ 7 = 285,71 токена на параметр — в 14 раз больше оптимума Chinchilla в 140 млрд токенов.
Советы
- Закладывайте запас. Перезапуски после сбоев, оценка качества и чекпоинты обычно добавляют десятки процентов к идеальному времени.
- Память проверяйте отдельно. Вычисления говорят, сколько учить; поместится ли модель, зависит от параметров, градиентов и состояний оптимизатора — N поможет найти калькулятор параметров трансформера.
- Умножьте GPU-часы на цену часа одного GPU, чтобы оценить стоимость.