Module.to() 是一个“in-place”方法,tensor.to() 函数不是。tensor.to() 这个函数功能是产生一个新的tensor,并不会改变原数据。
注意是否使用了tensor=tensor.to(device)
Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
最新推荐文章于 2024-06-06 18:57:42 发布
Module.to() 是一个“in-place”方法,tensor.to() 函数不是。tensor.to() 这个函数功能是产生一个新的tensor,并不会改变原数据。
注意是否使用了tensor=tensor.to(device)