在将模型实例放到GPU上时,即用model.cuda()时,只会将__init__中的有self前缀的属性及函数放到GPU上,对于其他的函数无作用。所以非_init__中定义的函数和数据需要单独的放入GPU语句。
.cuda()相关
最新推荐文章于 2023-11-21 15:35:23 发布
在将模型实例放到GPU上时,即用model.cuda()时,只会将__init__中的有self前缀的属性及函数放到GPU上,对于其他的函数无作用。所以非_init__中定义的函数和数据需要单独的放入GPU语句。