直接就用cpu训练，把我的电脑cpu给爆了 >_<
chenxs1427 opened this issue · 2 comments
chenxs1427 commented
为啥不默认GPU的
Thresh-Frazier commented
尝试下在train.py中修改：
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
2212240289 commented
试下在train.py中修改： device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
这行代码放在哪里呀，没学过这些