今天在把.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、将列表当做堆栈来使用 堆栈数据结构实现的思想就是:'先进...