TensorFlow vs PyTorch в 2021

fgfg Picture

Какой из фреймворков эффективнее? Разобрался ведущий разработчик группы компьютерного зрения MTS AI Дмитрий Гончаров.

Сравнение фреймворков глубокого обучения​

Дмитрий Гончаров, ведущий разработчик группы CV в команде MTS AI, провел эксперимент, сравнив производительность и другие качественные характеристики популярных фреймворков глубокого обучения TensorFlow и PyTorch.

Picture

В статье на Habr он рассказал о разнице двух программ и своем опыте работы с ними. Дмитрий как активный пользователь TensorFlow и PyTorch решил сравнить скорость обучения (на GPU), скорость инференса (на GPU и на CPU), а также количество выделяемой видеопамяти при обучении и инференсе.

Для сравнения было выбрано несколько классических архитектур свёрточных нейронных сетей, которые отличаются размером, количеством и типом свёрточных слоёв и функций активации. В рамках эксперимента разработчик варьировал наиболее часто изменяемые параметры: размер батча — количество одновременно обрабатываемых нейронной сетью изображений — и размер входных изображений.

Конечная цель эксперимента — измерить среднее время шага обучения/инференса. Для этого могут использоваться любые данные, например, сгенерированные случайным образом.

Что показал эксперимент? Изучить подробные измерения и результат сравнения можно в оригинальной статье на Habr.

Новости
Главное за последнее время
Смотреть все
События
Узнайте больше о генеративном ИИ в промышленности на ЦИПР
ВобъективеИИ
Предсказания Маска и конкуренция на ИИ-поле
Новости команды
MTS AI подписала декларацию об ответственной разработке генеративного ИИ
MTS AI в медиа
Облачное видеонаблюдение по цене чашки кофе
ВобъективеИИ
ИИ-ассистенты будущего от Google и умный антивирус из России
Кейсы
Чат-бот на базе генеративного ИИ для документооборота
Новости команды
Как нам удалось расширить команду в период конкуренции за кадры
Решения
Аудиобейджи с ИИ для ритейла и HoReCa
ВобъективеИИ
«Голодек» с GPT-4 и новый Atlas
События
Вебинар 14 мая, 11:00. Внедрение ИИ в бизнес-процессы промышленных компаний.
ВобъективеИИ
Sora в Голливуде и ИИ для большого футбола