Open MaySoon2020 opened 8 months ago
找到全部.cuda() 修改为 to('cpu')
.cuda()
to('cpu')
在M1芯片的GPU上训练,试试将源代码中的torch.device("cuda")替换成torch.device("mps")?
找到全部
.cuda()
修改为to('cpu')