摘要:
这篇文章主要介绍了pytorch中的model=model.to(device)使用说明,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教 这代表将模型加载到指定设备上。 其中,device=torch.device("cpu")代表的使用cpu,而device=tor 阅读全文
posted @ 2021-07-10 22:22
linux-123
阅读(1285)
评论(0)
推荐(0)
浙公网安备 33010602011771号