model = Network()
model.to(device)
RuntimeError: Tensor for argument #2 ‘weight’ is on CPU, but expected it to be on GPU (while checkin
最新推荐文章于 2024-03-20 10:53:46 发布
model = Network()
model.to(device)