pytorch
DyingCZ
有机器学习和深度学习兴趣的小伙伴可以加我qq505632696,一起学习
展开
-
pytorch入门完整例子(包括建立模型,保存模型,恢复加载模型)
是用jupyter notebook写的,没安装的请自行安装地址:https://pan.baidu.com/s/1N9xMBwAv_outc6_1dbjhLQ原创 2019-01-10 21:13:51 · 2235 阅读 · 0 评论 -
pytorch卷积神经网络的全连接层第一层参数(input_features)设置
在用pytorch写网络结构的时候,会发现在卷积层与第一个全连接层的全连接层的input_features不知道该写多少? 这个时候我们可以换一个思路,由于pytorch支持动态图,所以我们可以把全链接层写在forward这个函数里面,这样就很好的解决了这个尴尬的问题。代码如下import torch as tcfrom torch import nnclass CNN(nn.M...原创 2019-01-09 21:54:32 · 11357 阅读 · 7 评论