Skip to content

[TVM] Tuning

Valentina Kustikova requested to merge github/fork/Rodimkov/tvm_tuning into master

Created by: Rodimkov

Первая версия настройки нейронной сети средствами TVM. Необходимо будет вынести параметры запуска, которые можно менять, подправить внешний вид. Сделаю это на последнем этапе, когда будет понятно остальное.

  • autotvm
  • auto_scheduler
  • meta_schedule
  1. Нужно ли написать парсер для аргументов командной строки как в остальных скриптах?
  2. Как данная часть будет запускаться? Что выносить в парсер?
  3. В текущей реализации предполагается, что модель и параметры модели формата TVM, нас ведь интересует именно такой вариант? Иначе необходимо писать реализацию конвертации модели из разных фреймворков. Будет ли часть с инференсом сохранять модели? Возможно, будет необходимо реализовать данную часть отдельно, но тогда будет логичным, чтобы часть с инференсом поддерживала представление моделей в формате TVM.
  4. Так как я мало знаком с DLI мне не очевидно какое дать название классам и методам :)

Merge request reports