今天在把.pt文件转ONNX文件时，遇到此错误。
报错
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_mm)
原因
代码中的Tensor，一会在CPU中运行，一会在GPU中运行，所以最好是都放在同一个device中执行。
pytorch有两种模型保存方式：
一、保存整个神经网络的的结构信息和模型参数信息，save的对象是网络net
二、只保存神经网络的训练模型参数，save的对象是net.state_dict()
对应两种保存模型的方式，pytorch也有两种加载模型的方式。对应第一种保存方式，加载模型时通过torch.load('.pth')直接初始化新的神经网络对象；对应第二种保存方式，需要首先导入对应的网络，再通过net.load_state_dict(torch.load('.pth'))完成模型参数的加载。
相关文章
我用Python修改了前男友的开机密码，登录后居然发现了一个秘密
Original: https://www.cnblogs.com/pythonQqun200160592/p/15704900.html Author: python可乐编程 Title: 我用Py...
ModuleNotFoundError: No module named ‘pyecharts’
ModuleNotFoundError: No module named 'pyecharts' 解决： pip install pyecharts Original: https://www.cnb...
python os模块详细用法
os 模块提供了非常丰富的方法用来处理文件和目录 # 获取当前路径 path = os.getcwd() # 获取当前绝对路径 os.path.abspath(path) # 创建一级目录 os.mk...
Python 输入/输出
程序要实现人机交互功能，需要能够向显示设备输出有关信息及提示，同时也要能够接收从键盘输入的数据。Python提供了用于实现输入/输出功能的函数input( )和输出函数 print（），下面分别对这两...
python—关闭 close，quit
python---关闭 close，quit 原创 夕陌2022-07-19 11:26:02©著作权 文章标签 导包 firefox 文章分类 Python 编程语言 ©著作权归作者所有：来自51C...
python自动化办公
一，xlwings的安装与使用 1，xlwings是什么 在日常生活中我们或多或少的都会跟Excel打交道，比如做销售统计，人力的考勤，学生的考试成绩等等，甚至在某些领域会涉及到批量操作Excel表格...
python之文件处理
前戏 应用程序运行过程中产生的数据最先都是存放于内存中的，若想永久保存下来，必须要保存于硬盘中。应用程序若想操作硬件必须通过操作系统，而文件就是操作系统提供给应用程序来操作硬盘的虚拟概念，用户或应用程...
Python 转义字符、格式化字符
Python 转义字符、格式化字符 精选原创 lanxiaofang2022-08-22 23:25:28博主文章分类：Python ©著作权 文章标签 python 转义 格式化 字符 文章分类 P...
7 天学个Go，Go 结构体 + Go range 来学学
写在学习前 在前一篇博客我们学习了 Go 数组，其要求所有元素为同一数据类型，如果希望存储不同类型的数据，就要用到结构体相关知识。 结构体的定义：存储相同或不同类型的数据集合。 有 C 相关经验，结构...
Python 协程并发下载斗鱼美女图片
仅用学习参考，转载请注明出处 爬虫技术第一步 作为爬虫技术第一步，就是可以使用爬取下载图片作为示例。 那么爬取图片，我们肯定去爬取美女图片作为第一步操作示例呀。 那么下面一个最重要的工作。 什么是最重...
初识Python系列（一）
对于Python selenium操作的总结（一） 1.对于驱动的安装 驱动包:webdriver（在cmd执行help(webdriver)可查看所支持的浏览器类型，在此只提其中的Chrome和Ed...
苏彤，你的 Python Flask 编写生成二维码接口写完了
需求背景 本篇博客实战解决一个真是场景，需求描述如下： 为团队其它成员提供一个二维码生成接口，传入文本，返回图片地址，数据存储到 OSS 中。 采用 Python Flask 框架实现，并且使用一个第...
说我菜？那好，我用Python制作电脑与手机游戏脚本来赢你
Original: https://www.cnblogs.com/pythonQqun200160592/p/15556966.html Author: python可乐编程 Title: 说我菜？...
万能的list列表，python中的堆栈、队列实现全靠它！
在python编程中很多的数据计算都需要依靠列表来实现，比如Java编程中的堆栈结构/队列结构也可以通过list列表实现的。 【阅读全文】 1、将列表当做堆栈来使用 堆栈数据结构实现的思想就是：'先进...