标签：input=input.to(device)

[mcj]pytorch中model=model.to(device)用法

2019-11-21马春杰杰评论(2)赞(0)

这代表将模型加载到指定设备上。其中，device=torch.device("cpu")代表的使用cpu，而device=torch.device("cuda")则代表的使用GPU。当我们指定了设备之后，就需要将模型加载到相应设备中，此...

马春杰杰博客始建于2017年6月14日，博客主要分享网站建设中遇到的问题及解决方案、自己在读书过程中的心得体会及分享一些自觉有意义的音视频内容，记录些生活中的琐事，希望博客能督促怠惰的自己不断学习，不断进步。

不公开

※ 微博 ※

不公开

※ 支付宝好友 ※

不公开

※ 个人QQ号 ※

精彩直达

友情链接视频解析简历生成表白网页联系我们给我留言

QQ咨询
QQ咨询
回顶
回顶部