pytorch
文章平均质量分 93
yanhe156
这个作者很懒,什么都没留下…
展开
-
pytorch的基本使用
需要特别注意的一点很多博客中提到了pytorch中的tensor基本上和numpy中的array是一样的,可以直接通过函数相互转化,同时pytorch中有variable变量,variable是对tensor的一个封装,用于构建计算图,并进行迭代。但是对0.4.1版本来说,Variable不再被支持使用。可以理解为Variable合并到了Tensor中。 我开始一些博客和文档同时看,一直以为t...原创 2018-10-05 16:47:15 · 2088 阅读 · 0 评论 -
pytorch 使用笔记
参数初始化自定义网络后,参数是有默认的初始化的,打印一下,可以看到不为0,查看源码可以看到卷积层默认使用的uniform初始化。似乎一般使用xavier初始化效果更好。比如为Conv2d类型的层进行参数初始化,初始化为常量。def weights_init(m): if isinstance(m, nn.Conv2d): nn.init.constant_(m.w...原创 2018-10-26 23:29:07 · 314 阅读 · 0 评论