摘要: 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 阅读(336) 评论(0) 推荐(0) 编辑