1: .cpu() 2: torch.set_default_tensor_type('torch.cuda.FloatTensor')
RuntimeError: ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor
最新推荐文章于 2023-03-02 10:36:31 发布
1: .cpu() 2: torch.set_default_tensor_type('torch.cuda.FloatTensor')