这代表将模型加载到指定设备上。 其中,device=torch.device('cpu')代表的使用cpu,而device=torch.device('cuda')则代表的使用GPU。
確定! 回上一頁