原因:变量没有加cuda

 

data=data.cuda()

posted on 2019-09-03 17:19  fuge92  阅读(6493)  评论(0编辑  收藏  举报