Pytorch_DeepFM icon indicating copy to clipboard operation
Pytorch_DeepFM copied to clipboard

Published 20 hours ago • verified publisher icon Hyfred

bias未设置device,导致use_cuda=True时报错

Open NIck-Meng opened this issue 3 years ago • 0 comments

bias未设置device,导致use_cuda=True时报错 RuntimeError: expected device cuda:0 but got device cpu 可以修改为： bias = torch.nn.Parameter(torch.randn(Xi.size(0))).to(self.device)

NIck-Meng avatar Jun 01 '21 08:06 NIck-Meng