TensorFlow vs PyTorch в 2021

Picture

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

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

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

Picture

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

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

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

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

Новости
Главное за последнее время
Смотреть все

Инвестиции

MTS AI в медиа

Новости команды

Партнерство

Кейсы

Решения

ВобъективеИИ

События

Технологии

Без рубрики

ВобъективеИИ
О порядке в данных, законах физики и видении сквозь стены
ВобъективеИИ
Бан LLM и зеленый свет автономному транспорту
Новости команды
В России подготовили первый курс разметки данных для людей с аутизмом
MTS AI в медиа
МТС защитил свыше 5 млн абонентов от спам-звонков с помощью ИИ 
ВобъективеИИ
ИИ на атомном и космическом уровне
События
Вебинар: реальные кейсы использования платформы синтеза и распознавания речи от MTS AI
ВобъективеИИ
Интервью с чат-ботом и ИИ-система для старения за секунды
MTS AI в медиа
Как и над чем работают в MTS AI и VisionLabs. Репортаж из офисов компаний
ВобъективеИИ
Как ИИ судит в футболе и за что судят ИИ
ВобъективеИИ
Искусный переговорщик, хороший диагност и так себе кулинар