MCPcopy Create free account
hub / github.com/OpenPPL/ppq / Example_Torch2trt.py

File Example_Torch2trt.py

ppq/samples/TensorRT/Example_Torch2trt.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1# ---------------------------------------------------------------
2# 这个脚本向你展示了如何使用 torch2trt 加速 pytorch 推理
3# 截止目前为止 torch2trt 的适配能力有限，不要尝试运行特别奇怪的模型
4# 你可以把模型分块来绕开那些不支持的算子。

Callers

nothing calls this directly

Calls 3

evalMethod · 0.80
stepMethod · 0.80
forwardMethod · 0.45

Tested by

no test coverage detected