PyTorch
ycszen
书山有路勤为径
展开
-
PyTorch参数初始化和Finetune
前言这篇文章算是论坛PyTorch Forums关于参数初始化和finetune的总结,也是我在写代码中用的算是“最佳实践”吧。最后希望大家没事多逛逛论坛,有很多高质量的回答。参数初始化参数的初始化其实就是对参数赋值。而我们需要学习的参数其实都是Variable,它其实是对Tensor的封装,同时提供了data,grad等借口,这就意味着我们可以直接对这些参数进行操作赋值了。这就是PyTorch简洁原创 2017-03-24 19:45:13 · 18436 阅读 · 9 评论 -
PyTorch预训练
前言最近使用PyTorch感觉妙不可言,有种当初使用Keras的快感,而且速度还不慢。各种设计直接简洁,方便研究,比tensorflow的臃肿好多了。今天让我们来谈谈PyTorch的预训练,主要是自己写代码的经验以及论坛PyTorch Forums上的一些回答的总结整理。微改模型PyTorch中的torchvision里已经有很多常用的模型了,可以直接调用:AlexNetVGGResNetS原创 2017-03-24 17:12:16 · 5194 阅读 · 0 评论