@MinZhaoLin 你遇到的问题好像和这个issue不一样
metric里传入的pred_label是GPU数据,label是CPU数据
你可以尝试pred_label.as_in_context(label.context),让pred和label在同一个context里

[ Full content available at: 
https://github.com/apache/incubator-mxnet/issues/9920 ]
This message was relayed via gitbox.apache.org for [email protected]

Reply via email to