我遇到这个问题的原因是,数据已经放到GPU上,但是模型还在Cpu上,因此会遇到这个问题. 解决方法就是,先把模型也放到GPU上,然后在把放到GPU上的数据输入到模型中.
更多推荐
Unable to get repr for <class torch.Tensor>
>www.elefans.com
编程频道|电子爱好者 - 技术资讯及电子产品介绍!
发布评论