Преобразование модели Pytorch с ненулевыми операциями в TensorRT

#pytorch #converters #onnx #tensorrt

Вопрос:

Я пытаюсь преобразовать модель Pytorch, содержащую ненулевые операции, в TRT.

Я успешно сохранил его как ONNX, но ни TensorRT не поддерживает эту операцию изначально, ни существующие преобразователи, такие как https://github.com/onnx/onnx-tensorrt

Не могли бы вы посоветовать, как я могу справиться с этим, возможно, изменив операцию на » Где » или «Равно Логически нет»?

Комментарии:

1. Что такое нарушающая функция и как вы ее используете?

2. Извините, что вы имеете в виду под функцией оскорбления?

3. Я пытаюсь преобразовать модель CrowdDet github.com/Purkialo/CrowdDet

4. Пожалуйста, опубликуйте свою трассировку ошибок стека и, в частности, функцию «нарушение» (что означает возникновение ошибки).