Skip to content

[TVM] retinaface-resnet50-pytorch NotImplementedError #458

Open

Description

@ismukhin

@n-berezina-nn, добрый день! Возникла проблема с моделью retinaface-resnet50-pytorch при запуске скрипта inference_tvm_pytorch.py выбрасывается следующая ошибка:

[ ERROR ] Traceback (most recent call last):
  File "/home/vanya/projects/dl-benchmark/src/inference/./inference_tvm_pytorch.py", line 181, in main
    graph_module = converter.get_graph_module()
  File "/home/vanya/projects/dl-benchmark/src/model_converters/tvm_converter/tvm_converter.py", line 75, in get_graph_module
    model = self._convert_model_from_framework(target, dev)
  File "/home/vanya/projects/dl-benchmark/src/model_converters/tvm_converter/tvm_converter.py", line 146, in _convert_model_from_framework
    model, params = self.tvm.relay.frontend.from_pytorch(scripted_model, shape_list)
  File "/home/vanya/anaconda3/envs/dl_benchmark_env/lib/python3.9/site-packages/tvm/relay/frontend/pytorch.py", line 4979, in from_pytorch
    converter.report_missing_conversion(op_names)
  File "/home/vanya/anaconda3/envs/dl_benchmark_env/lib/python3.9/site-packages/tvm/relay/frontend/pytorch.py", line 4053, in report_missing_conversion
    raise NotImplementedError(msg)
NotImplementedError: The following operators are not implemented: ['aten::__isnot__', 'aten::smooth_l1_loss', 'prim::Exit', 'aten::__contains__', 'aten::uniform_', 'aten::_set_item', 'aten::insert', 'aten::conv2d', 'aten::__is__', 'prim::dtype', 'aten::items', 'prim::CreateObject', 'aten::list', 'aten::dim', 'aten::dict', 'prim::type', 'prim::tolist', 'prim::max', 'aten::__range_length', 'prim::Uninitialized', 'aten::warn', 'aten::__derive_index', 'aten::extend', 'prim::TupleIndex', 'aten::format', 'prim::Enter', 'aten::copy_', 'prim::unchecked_cast', 'aten::as_tensor', 'aten::keys', 'aten::item', 'aten::binary_cross_entropy_with_logits', 'prim::DictConstruct', 'aten::unique_dim', 'aten::values', 'aten::floordiv', 'prim::SetAttr', 'aten::append', 'prim::min']

Ошибка говорит о том, что TVM на низком уровне не поддерживает данные операторы, поэтому в настоящее время сконвертировать модель в представление TVM не получится. На какую другую модель можно заменить данную модель?

Activity

  1. n-berezina-nn commented on Feb 5, 2024

    @n-berezina-nn
    Collaborator

    @ismukhin @valentina-kustikova эта проблема еще актуальна?

  2. valentina-kustikova commented on Feb 6, 2024

    @valentina-kustikova
    Contributor

    @ismukhin @valentina-kustikova эта проблема еще актуальна?

    Да, проблема с запуском актуальна. В почте по итогам договорились использовать модели детектирования, которые точно поддерживает TVM согласно документации, они присутствуют в результатах валидации.

  3. n-berezina-nn commented on Feb 12, 2024

    @n-berezina-nn
    Collaborator

    Можем закрыть этот тикет, если модель заменили?

  4. valentina-kustikova commented on Feb 13, 2024

    @valentina-kustikova
    Contributor

    Можем закрыть этот тикет, если модель заменили?

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions