mxnet.base.MXNetError: src/imperative/./imperative_utils.h:70: Check failed: inputs[i]->ctx().dev_mask() == ctx.dev_mask() (1 vs. 2)
2024-09-05 19:12:36
mxnet 训练错误：
mxnet.base.MXNetError: [14:42:22] src/imperative/./imperative_utils.h:70: Check failed: inputs[i]->ctx().dev_mask() == ctx.dev_mask() (1 vs. 2) Operator stack require all inputs live on the same context. But the first argument is on gpu(0) while the 2-th argument is on cpu(0)
这种错误就是数据存储不同导致的，前一个数据在gpu上面，后一个数据在cpu上面，导致错误异常。
修改方法：在某一个数据加上context限制，使用ctx=mx.gpu()或者ctx=mx.cpu()
最新文章
- <<< Java提取网页源码
- eclipse提交项目到github
- Greenplum 生成加分区语句
- ListView视图缓存错位问题
- Java—static、this、super用法总结
- Linux 创建swap分区
- windows内存管理方式以及优缺点
- HDN2048（交错复发）
- MySQL中函数、游标、事件、视图基本应用举例（代码）
- 201521123082《Java程序设计》第4周学习总结
- lintcode.66 二叉树前序遍历
- HTTP协议详解（四）
- Eclipse 中 Maven 项目默认JDK版本为1.5 的解决方法
- 『关键点检测』CPN：Cascaded Pyramid Network for Multi-Person Pose Estimation
- Win10 + VS2017 15.5.6 环境下解决 Python 3.6 环境无法刷新DB的问题
- 微信小程序跳转分析
- personal的制作
- C#new出来的结构体内存分配在堆上
- 我进行jvm内存调优的一些记录
- 【struts2】继承ActionSupport类