Модель на 0,5 млрд параметров обучили на одной RTX 4090 за 6 часов с fp8

Обучить языковую модель с нуля за обеденный перерыв? Ещё год назад это звучало как фантастика. Сборка кластера из сотен видеокарт, бюджеты уровня небольшого стартапа и недели ожидания — вот что ассоциировалось с тренировкой ИИ. Но появление инструментов вроде nanochat от Андрея Карпатого и поддержка вычислений в формате fp8 перевернули всё. Я провел эксперимент: заставил одну-единственную RTX 4090 за 6 часов создать модель на 0.5 миллиарда параметров, обученную на русскоязычных IT-статьях с Хабра. Спойлер: это сработало. И работает на удивление стабильно.

Почему это вообще возможно? Разрушаем мифы

Раньше главным ограничением была память. Модели «весили» гигабайты, и чтобы их учить, нужны были промышленные карты с 80 ГБ VRAM. Формат fp8 — это хитрость, которая режет «вес» каждого параметра вдвое. Представьте, что вы записываете числа не с точностью до 16 знаков, а всего до 8. Звучит как потеря данных, да? На практике — потеря менее 0.3% в точности, но экономия 40% времени и 30% памяти. Это как перейти с чемодана на рюкзак: вещи те же, но нести легче.

Собираем датасет: Хабр как источник мудрости

Первый шаг — добыть «сырьё». Я не стал мудрить и взял API Хабра. Простой скрипт на Python за 15 минут скачал 5000 статей из разделов «Программирование», «Научпоп» и «IT-инфраструктура». Важный нюанс: не тащите весь текст с HTML-тегами. Чистите. Оставляйте только голый текст и переносы строк. Иначе модель выучит, что

— это важная часть языка. Полученный файл я разбил на чанки по 2048 токенов. Это стандартный «кусочек» для GPT-подобных моделей — достаточно, чтобы уловить контекст, но не перегрузить память.

Как работает nanochat: минимализм, который экономит часы

Nanochat — это форк знаменитого nanoGPT, но заточенный под диалоги. Он берёт на себя всю рутину: от токенизации до сохранения чекпоинтов. Для модели на 0.5B параметров я использовал конфиг, похожий на GPT-2 Medium: 12 слоёв, размер эмбеддинга 768. Ключевое отличие — поддержка fp8. В файл конфига я добавил всего две строки: model[“precision”] = “fp8” и optimizer[“dtype”] = “fp32”. Это значит, что веса модели хранятся в точном 32-битном формате, а вот градиенты (исправления ошибок) — в сжатом 8-битном. Именно это даёт ускорение.

Сравнение fp8 vs bf16: цифры, которые не врут

Я запустил обучение дважды: сначала в классическом bf16, потом в fp8. Результаты — в таблицу. Время: 3 часа 12 минут против 1 часа 55 минут. Размер батча (сколько примеров модель видит за раз): 16 против 24. Потребление VRAM: 11.5 ГБ против 8 ГБ. А качество? Перплексия (показатель «уверенности» модели) изменилась с 4.78 до 4.80. Для тех, кто не в теме: это настолько мизерная разница, что её не заметит даже опытный инженер. Субъективно — обе модели пишут осмысленные ответы на вопросы вроде «Как оптимизировать код на Python?». Разница — в скорости и возможности впихнуть больше данных за тот же бюджет.

Личное наблюдение автора: страх перед «сжатием»

Когда я впервые запустил обучение в fp8, то боялся, что модель «сойдёт с ума». В интернете полно страшилок, что низкая точность ломает градиенты и ИИ начинает галлюцинировать. На практике — ничего подобного. Градиентный скейлер (встроенная защита от переполнения чисел) и обрезка градиентов до 1.0 решили все проблемы. Модель училась стабильно, как локомотив по рельсам. Главный бонет, который я вынес: теперь я могу запустить эксперимент утром, а к обеду уже тестировать результат. Для быстрых прототипов fp8 — это новый стандарт. Для продакшена я бы оставил bf16, но для дома — идеальный компромисс.

Как повторить: инструкция для смелых

• Шаг 1. Клонируйте репозиторий nanochat и установите зависимости. Всё стандартно: git clone, pip install.
• Шаг 2. Скачайте датасет. Используйте скрипт из статьи (или возьмите готовый из репозитория проекта — 200 МБ, 5000 статей).
• Шаг 3. Настройте конфиг под fp8. Не забудьте включить grad_scaler и clip_grad_norm=1.0.
• Шаг 4. Запустите обучение. На RTX 4090 с 24 ГБ памяти это займёт около 2 часов.
• Шаг 5. Проверьте результат. Используйте скрипт chat.py и задайте модели любой IT-вопрос.

Коротко от автора

FP8 не заменит промышленные вычисления, но он делает ИИ-разработку демократичной. Теперь не нужно просить у начальства кластер, чтобы проверить гипотезу. Вы можете сделать это дома, на своей видеокарте, за чашкой кофе. И это меняет правила игры. Если вы всё ещё думаете, что обучение моделей — это удел гигантов, просто попробуйте. Время дешёвых экспериментов наступило.

По материалам: Tehnoseo