一、list[tensor,tensor]类型转tensor,报错,要求:list[tensor.numpy()]->tensor。此后tensor为cpu类型,要及时转为cuda
一、list[tensor,tensor]类型转tensor,报错,要求:list[tensor.numpy()]->tensor。此后tensor为cpu类型,要及时转为cuda
本文标题:pytorch经验1:cuda和cpu不能频繁变更
本文链接:https://www.haomeiwen.com/subject/txnloctx.html
网友评论