rknn-toolkit2
rknn-toolkit2 copied to clipboard
How to handle unimplemented Pytorch Tensor Operators using rknn-toolkit2 ?
When loading Pytorch AI Model using RKNN Toolkit2 , the following Error logs occurred.
How to fix the pytorch Exception?
How to replace unimplemented operators that RKNN doesn't support
Let us know how to implement custom Pytorch tensor Operators?
W init: rknn-toolkit2 version: 1.3.0-11912b58 --> Config model W config: 'target_platform' is None, use rk3566 as default, Please set according to the actual platform! done --> Loading model
E load_pytorch: The following operators are not implemented: ['aten::isnot', 'aten::dim', 'aten::kl_div', 'aten::is', 'aten::warn', 'prim::SetAttr', 'aten::append', 'aten::randint', 'prim::max', 'aten::floordiv', 'aten::remainder', 'prim::min', 'prim::Print', 'prim::unchecked_cast', 'aten::conv1d', 'aten::scatter_', 'prim::Uninitialized', 'prim::dtype', 'aten::ctc_loss', 'aten::item', 'aten::fill_', 'aten::conv2d', 'aten::format', 'aten::tril'] W load_pytorch: ===================== WARN(2) ===================== E rknn-toolkit2 version: 1.3.0-11912b58 W load_pytorch: Catch exception when loading pytorch model: ValueError("The following operators are not implemented: ['aten::isnot', 'aten::dim', 'aten::kl_div', 'aten::is', 'aten::warn', 'prim::SetAttr', 'aten::append', 'aten::randint', 'prim::max', 'aten::floordiv', 'aten::remainder', 'prim::min', 'prim::Print', 'prim::unchecked_cast', 'aten::conv1d', 'aten::scatter_', 'prim::Uninitialized', 'prim::dtype', 'aten::ctc_loss', 'aten::item', 'aten::fill_', 'aten::conv2d', 'aten::format', 'aten::tril']",) E load_pytorch: Traceback (most recent call last): E load_pytorch: File "rknn/api/rknn_base.py", line 1310, in rknn.api.rknn_base.RKNNBase.load_pytorch E load_pytorch: File "rknn/base/convertor/torch2onnx/pytorch.py", line 4209, in rknn.base.convertor.torch2onnx.pytorch.from_pytorch E load_pytorch: File "rknn/base/convertor/torch2onnx/pytorch.py", line 3341, in rknn.base.convertor.torch2onnx.pytorch.PyTorchOpConverter.report_missing_conversion E load_pytorch: File "rknn/api/rknn_log.py", line 113, in rknn.api.rknn_log.RKNNLog.e E load_pytorch: ValueError: The following operators are not implemented: ['aten::isnot', 'aten::dim', 'aten::kl_div', 'aten::is', 'aten::warn', 'prim::SetAttr', 'aten::append', 'aten::randint', 'prim::max', 'aten::floordiv', 'aten::remainder', 'prim::min', 'prim::Print', 'prim::unchecked_cast', 'aten::conv1d', 'aten::scatter_', 'prim::Uninitialized', 'prim::dtype', 'aten::ctc_loss', 'aten::item', 'aten::fill_', 'aten::conv2d', 'aten::format', 'aten::tril']