摘要: 错误1: Expected object of backend CPU but got backend CUDA for argument #2 'weight' #之前: model = model.cuda() #之后: model = torch.nn.DataParallel(model). 阅读全文
posted @ 2020-07-08 16:18 Manuel 阅读(146) 评论(0) 推荐(0) 编辑