
机器学习
special_hang
这个作者很懒,什么都没留下…
展开
-
pytorch中的Optimizer的灵活运用
pytorch可以自动求导梯度,不过要对需要求梯度的tensor设置为“requires grad = TRUE"。同时,pytorch可以对神经网络进行灵活训练,主要同通过Optimazer的设计来实现的。以下针对不同的应用场景需要对optimizer的使用进行总结。1. 多个神经网络联合训练应用场景(1)Autoencoder网络分为encoder与decoder两部分组成,...原创 2019-04-29 18:33:09 · 8361 阅读 · 2 评论 -
Pytorch之搭建神经网络的四种方法
转发自这里利用pytorch来构建网络模型有很多种方法,以下简单列出其中的四种。假设构建一个网络模型如下:卷积层--》Relu层--》池化层--》全连接层--》Relu层--》全连接层首先导入几种方法用到的包import torchimport torch.nn.functional as Ffrom collections import OrderedDict第一种...转载 2019-04-29 18:50:50 · 518 阅读 · 0 评论