天池学习

新手解读trt2023(ControlNet)源码

0
0
0
文章

0


全部评论(2)
  • 膜拜大佬。。。

    2楼0回复发表于浙江省
  • 你好,我想问一下这个导入前的替换,具体是怎么样的?(之前没有做过onnx模型导出这方面的工作)

    1楼0回复发表于台湾省
    2023-07-17 17:19:37
    你问替换是指pytorch换tensorRT吗?基本就是model.forward 换成tensorRT engine推理。engine推理可以看示例代码仓库https://github.com/NVIDIA/trt-samples-for-hackathon-cn/tree/master/cookbook,里面有很多案例。
    0回复发表于广东省