解决PyTorch报错:RuntimeError: Expected object of backend CPU but got backend CUDA for argument #2 'other

1. 问题描述 在利用PyTorch报错:python RuntimeError: Expected object of backend CPU but got backend CUDA for argument #2 'other' 报错时,对应Python语句的写法为:web update_scale = torch.max(scale.round(), torch.Tensor([2]))
相关文章
相关标签/搜索