摘要: 包括两步: 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) 编辑