随笔分类 -  Pytorch

Pytorch使用心得
Pytorch使用-1
摘要：1. pytorch初体验 1. 有两类情况都必须执行 to(cuda)操作：(1)需要input到model的input_tensor, (2)model, e.g. 2. 有时候为了确保计算图，tensor，Autograd，的正常，优先使用pytroch提供的numpy api，因为直接使用原 阅读全文
posted @ 2019-04-03 20:34 hzhang_NJU 阅读(346) 评论(0) 推荐(0)