会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
yanghh
新随笔
管理
2020年12月10日
如何用 GPU 训练模型?
摘要: 包括两步: 1)Convert parameters and buffers of all modules to CUDA Tensor. 2)Send the inputs and targets at every step to the GPU. 注意:模型和数据要迁移到同一块显卡上。 举个例子
阅读全文
posted @ 2020-12-10 20:00 _yanghh
阅读(2221)
评论(0)
推荐(0)
编辑