会员
周边
众包
新闻
博问
闪存
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
linux-123
博客园
首页
新随笔
联系
订阅
管理
2021年7月10日
解说pytorch中的model=model.to(device)
摘要: 这篇文章主要介绍了pytorch中的model=model.to(device)使用说明,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教 这代表将模型加载到指定设备上。 其中,device=torch.device("cpu")代表的使用cpu,而device=tor
阅读全文
posted @ 2021-07-10 22:22 linux-123
阅读(1135)
评论(0)
推荐(0)
编辑
公告