arutema47's blog

書いたり書かなかったり。

2020-11-28から1日間の記事一覧

Pytorch高速化 (3) TensorRTで推論を10倍高速化

TLdr; torch2trtというpytorchモデルをTensorRTに簡単に変換するライブラリを使い、Jetson nano+xavier上で画像認識とセグメンテーションの推論処理を10倍高速化できることを確認しました。 ただtorch2trtはカスタムモデルには対応していないため(resnetなど…