RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be

RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be the same 如問題所示,你的輸入的類型是字節型的tensor,而加載的權重的類型是float類型的tensor,需要將字節型的tensor轉化爲float型的tensor。eg:我的程序的部分截
相關文章
相關標籤/搜索