pytorch
多读多写多思考
这个作者很懒,什么都没留下…
展开
-
pytorch :使用两次sort函数(排序)找出矩阵每个元素在升序或降序排列中的位置
对于一个二维矩阵,使用两次sort函数(对原矩阵、第一次sort之后的index)后,index(第二个变量)的输出是原矩阵一个由大到小(或由小到大)的映射,映射的值为整数,位置保持不变。...原创 2017-12-14 17:17:17 · 15275 阅读 · 3 评论 -
pytorch下分类神经网络的迁移学习transfer learning
对预训练模型的迁移引用【1】中的提法,分为两种形式只训练最后fc层的freeze and train以预训练模型为初始参数,训练所有层的finetune这里只讨论网络结构的变更finetunemodel_ft = models.resnet18(pretrained=True)num_ftrs = model_ft.fc.in_features #最后fc层的输入原创 2018-01-10 09:47:22 · 8188 阅读 · 0 评论 -
pytorch Module里的children()与modules()的区别
children()与modules()都是返回网络模型里的组成元素,但是children()返回的是最外层的元素,modules()返回的是所有的元素,包括不同级别的子元素。官方论坛的回答:Module.children() vs Module.modules()我以fmassa的举例为例:m = nn.Sequential(nn.Linear(2,2), ...原创 2018-01-09 20:20:10 · 25434 阅读 · 12 评论 -
visdom的安装及在pytorch下损失函数的可视化应用
2018/9/18更新 感觉tensorboardX插件更好用,已转用https://github.com/lanpa/tensorboardX 更新:新版visdom0.1.7安装方式为:conda install -c srivasv visdom pytorch下可采用visidom作为可视化工具1. 安装 pip install visdomconda ...原创 2018-01-10 08:37:30 · 43265 阅读 · 18 评论 -
pytorch的基础操作:Tensor, Variable
一、 Tensor1. 产生一个值未初始化的tensorimport torchx=torch.Tensor(5,3) 1.00000e-36 * 0.0000 0.0000 0.0000 0.0000 0.0000 0.0000 1.7644 0.0000 0.0000 0.0000 0.0001 0.0000 1.8972 0.0000 0.0000...原创 2018-04-28 09:22:11 · 1845 阅读 · 0 评论 -
pytorch 模型部分参数的加载
如果对预训练模型的结构进行了一些改动,在训练的开始前希望加载未改动部分的参数,如将resnet18的第一层卷积层conv1的输入由3通道改为6通道的new_conv1,将分类层fc的1000类输出改为2类输出的new_fc,注意:要改一下名字与原来的不同。导入模型myNet=resnet18()然后就加载模型的参数,参考pytorch 如何加载部分预训练模型pretrained_dict=torc...原创 2018-04-28 14:03:46 · 33564 阅读 · 14 评论 -
pytorch模型中间层特征的提取
参考pytorch论坛:How to extract features of an image from a trained model定义一个特征提取的类:#中间特征提取class FeatureExtractor(nn.Module): def __init__(self, submodule, extracted_layers): super(Featur...原创 2018-04-28 14:14:24 · 18751 阅读 · 21 评论