01 2022 档案
摘要：torchsummary的使用 使用流程安装导入使用 官方说明demo 建议查看官方demo --> github 使用流程 安装 pip install torchsummary 导入 from torchsummary import summary 使用 # 参数说明 summary(your_
阅读全文
摘要：libtorch处理多输入/多输出问题 准备工作多输入问题调整网络模型设置libtorch输入 多输出问题参考文章 一般处理多输入/多输出问题时，pytorch中容易处理；但是在libtorch中会出现一些问题。 这篇博客为记录用。 准备工作 首先加载多模型。 因为可能需要根据不同情况调用多个模型，
阅读全文
摘要：在 C++ 中加载 TorchScript 模型 推荐阅读官方文档： 如何保存模型 如何在c++中加载模型 pytorch的c++ api torchScript文档 以下内容基于官方文档写一些注释~~ 将pytorch模型转换为torch脚本 pytorch模型从python到c++是通过torc
阅读全文
摘要：Debug savept.py文件用来将gpu上训练的模型转换为cpu上推理可用的pt文件 以下为部分代码： import torch from model import NET print(torch.__version__) # 1.10.1+cpu model = NET device = t
阅读全文
摘要：保存和加载模型 只保存模型的参数 保存 torch.save(model.state_dict(),'xxx.pth') 加载 model = net() #首先要先定义网络模型 state_dict = torch.load('xxx.pth') # 读取pth文件中的参数 model.load_
阅读全文
