- 博客(1)
- 收藏
- 关注
原创 Pytorch学习笔记
问题1:加载模型需要指定GPU或者CPUstate = torch.load(path, map_location=‘cuda:0’)问题2:torch.randn与torch.rand区别torch.rand(sizes, out=None) → Tensor 均匀分布torch.randn(sizes,out=None) → Tensor 标准正太分布torch.normal(me...
2020-01-14 14:53:15 233
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人