[TVM] retinaface-resnet50-pytorch NotImplementedError #458
Description
@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 не получится. На какую другую модель можно заменить данную модель?