- 博客(4)
- 收藏
- 关注
RSS订阅原创 pytorch在fintune时将sequential中的层输出，以vgg为例
pytorch将sequential中的层输出，以vgg为例有时候我们在fintune时发现pytorch把许多层都集合在一个sequential里，但是我们希望能把中间层的结果引出来做下一步操作，于是我自己琢磨了一个方法，以vgg为例，有点僵硬哈！首先pytorch自带的vgg16模型的网络结构如下：VGG( (features): Sequential( (0): ...
2018-04-15 16:15:57
5105
4
原创 pytorch在不同的层使用不同的学习率
pytorch在不同的层使用不同的学习率有时候我们希望某些层的学习率与整个网络有些差别，这里我简单介绍一下在pytorch里如何设置，方法略麻烦，如果有更好的方法，请务必教我：首先我们定义一个网络：class net(nn.Module): def __init__(self): super(net, self).__init__() sel...
2018-04-15 15:41:51
21418
4
原创 pytorch本来正常使用的代码突然out of memory
最近在在一台新的服务器上安装了pytorch，将之前的网络移植过来跑却out of memory了，仔细看了代码没有发现什么问题。最后发现是安装的pytorch版本太低了。首先查看pytorch版本：import torchprint (torch.__version__)我发现安装的是0.1.12版本的，与现在不符合，需要更新pytorch。我是用conda更新pytorchconda
2017-12-29 10:51:02
6355
原创 Linux上用anaconda安装pytorch
Linux上用anaconda安装pytorchPytorch是一个非常优雅的深度学习框架。使用anaconda可以非常方便地安装pytorch。下面我介绍一下用anaconda安装pytorch的步骤。首先，我们需要下载anaconda，在https://www.anaconda.com/download/上可以选择对应python版本的anaconda。下载好后通过运行下载好的.sh文件安装a
2017-12-27 20:42:53
21028
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人