Outpost VFX ускорила обучение моделей для замены лиц в 8 раз с помощью AWS

Outpost VFX, работающая со студиями в Великобритании, Канаде и Индии, сообщила, что ускорила обучение AI-моделей для замены лиц в 8 раз после перехода на инфраструктуру AWS. Как пишет AWS Machine Learning Blog, компания обновила workflow для визуальных эффектов, чтобы сократить задержки в производстве и быстрее проходить циклы согласования с клиентами.

Раньше студия обучала модели на рабочих станциях с RTX 3090 и использовала небольшие датасеты с актёрами и их дублёрами. По данным компании, дообучение одной модели занимало от 1 до 2 недель. Традиционные процессы композитинга и de-aging при этом требовали более 5 дней только на подготовку первых версий для режиссёрского утверждения.

Чтобы масштабировать обучение, Outpost VFX вместе с AWS Generative AI Innovation Center адаптировала код модели под распределённое обучение на нескольких GPU. Решение развернули на Amazon EC2 P5 instances с GPU NVIDIA H100 в изолированной облачной среде, которая соответствует требованиям безопасности компании. В Outpost VFX также отметили, что архитектуру рассчитывали с учётом работы с более крупными датасетами и изображениями более высокого разрешения.