pytorch
summer>
move on !
展开
-
pytorch那些事(1)
笔记小汇总,详情参照https://pytorch-cn.readthedocs.io/zh/latest/notes/autograd/1.从后向中排除子图每个变量有requires_grad和volatile两个标志,允许从梯度计算中精细地排除子图如果所有变量都不需要梯度,后向计算不在子图中执行,requires_grad可以冻结部分模型;volatile决定requires_grad为fal...原创 2018-04-10 20:22:14 · 457 阅读 · 0 评论 -
pytorch那些事(2)
笔记小汇总,详情参照:https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/1.torch包:有多维张量的数据结构及基于其上的多种数学操作;提供多种工具,更有效对张量和任意类型序列化;有CUDA的对应实现,可在NVIDIA GPU上进行张量运算2.张量Tensorstorch.is_tensor(obj):判断对象...原创 2018-04-12 15:29:11 · 4575 阅读 · 0 评论