@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]
