摘要:
包括两步: 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
阅读(2245)
评论(0)
推荐(0)
浙公网安备 33010602011771号