参考: #6162 #9725
原因:模型和输入数据所在设备不匹配。
In fact, MMEngine can automatically change your module into cuda mode i.e. model = model.to(get_device()) in runner.py. Please double check your custom implementation like wrongly using list instead of modulelist for injection.
已经说的很清楚了,MMEngine 会自动调整为 cuda 模式。所以不要再说什么 model.cuda
等等手动方法了。应该是模型定义中出的问题。
注意检查模块实现的方法,应当使用 nn.Module
而不是 list
或 nn.Sequential
。
将代码中用到 list
的部分替换,报错解决。