wkcn commented on issue #9920: Asynchronous Issue on CustomOP and mxnet.image.ImageDetIter URL: https://github.com/apache/incubator-mxnet/issues/9920#issuecomment-415599500 @MinZhaoLin 你遇到的问题好像和这个issue不一样 metric里传入的pred_label是GPU数据,label是CPU数据 你可以尝试pred_label.as_in_context(label.context)
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services