Skip to content

MXNet inference

Created by: valentina-kustikova

Реализация вывода средствами библиотеки MXNet. Добавлена поддержка вывода классификационных моделей через Gluon API из Gluon Model Zoo, а также загрузка моделей, которые были обучены вручную и экспортированы в файлы. Модели должны содержать 1 вход и 1 выход, точность - FP32.

Задачи в отдельные PR:

  1. Поддержка int8-моделей (нужна версия 1.9.*)
  2. Поддержка неклассификационных моделей (нужна версия 1.9.*)
  3. Dockerfile для запуска массовых экспериментов
  4. Поддержка проверки точности работы моделей (обновить AccuracyChecker)
  5. Обновление wiki
  6. Обновление GUI-приложения для формирования конфига MXNet

Merge request reports