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