// torch::jit::script::Module module;
// module = torch::jit::load("/media/ljt/0A9AD66165F33762/LOGPATH/20201206_065944_lr_4e-04_b_8/model2021.pt");
// module.eval();
// //torch::Tensor tensor2 = torch::rand({1, 3, 288, 800});
// std::vector<torch::jit::IValue> inputs;
// inputs.emplace_back(torch::rand({1, 3, 288, 800}));
// torch::Tensor out_tensor = module.forward(inputs).toTensor();
// std::cout << out_tensor.slice(/dim=/1, /start=/0, /end=/200) << '\n';
cpp
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- 跟着网上(百度)的教程做libtorch部署,官方库正在更新,有很多问题没有解答,Google了一下把解决的问题列...
- 用pytorch训练了安全帽检测的二分类模型,训练的模型在python中效果还不错,现在想通过libtorch在C...
- torch 本笔记引用自PyTorch中文文档 包torch包含了多维疑是的数据结构及基于其上的多种数学操作。 1...
- 点击访问:PyTorch中文API应用具体代码地址 自动求导机制 本说明将概述Autograd如何工作并记录操作。...