libtorch推理速度过慢

1 原因

torchscript runtime在第一阶段会进行优化

2 解决办法

#include <torch/csrc/jit/python/update_graph_executor_opt.h>

torch::jit::setGraphExecutorOptimize(false);

参考:https://discuss.pytorch.org/t/speed-of-custom-rnn-is-super-slow/63209

posted @   珠峰上吹泡泡  阅读(238)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
历史上的今天:
2019-05-03 分类整理
点击右上角即可分享
微信分享提示