pytorch
luissen
Graduate student in Peking university, major in computer vision, especially object detection, super resolution and domain adaption reid.
展开
-
在linux客户端使用PIP安装pytorch0.4
在pytorch官网上选择linux版本以及对应的python版本与cuda版本,自动生成语句。(记得选择pip)然后再服务器客户端中输入对应语句即可(有的pip3要改成pip,不然会报错)注意:若装了0.3版本的pytorch,安装0.4会把原来的0.3卸载。...原创 2018-10-29 18:33:11 · 7527 阅读 · 0 评论 -
pytorch错误ImportError: dlopen: cannot load any more object with static TLS
今天运行pytorch文件时出现了这个错误经过多次排查,发现是因为把import torch放在了import cv2下面了如果要用opencv,把import torch 放在import cv2前面就可以了...原创 2018-11-16 17:17:34 · 8132 阅读 · 3 评论 -
Pytorch只加载自己需要的模型参数(修改模型后)
给定一个预训练模型,如果你对模型结构做了一定的修改,那么可以只加载未改变的模型参数,从而加快模型的训练。代码如下:pretrained_dict = ‘…….pkl’#预训练模型参数保存地址model_dict = model.state_dict() #自己的模型参数变量pretrained_dict = {k: v for k, vin pretrained_dict.items() i...原创 2018-12-16 22:57:19 · 4898 阅读 · 1 评论 -
【pytorch】计算一个CNN模型model的参数量,即论文中常用的#parameters
在开始之前,请记住,存储量不是参数量!!!!存储量不是参数量!!!!存储量不是参数量!!!!通常来说我们在训练模型的时候会用checkpoint的方法把模型保存下来,一个模型小则几十M,大则上百M,并且我们很多时候会把这个存储量误认为是参数量,比如但是实际上这个是存储量而不是参数量,那么在pytorch中参数量怎么计算呢?实际上我们可以直接在pytorch代码中print一个模型的参数:...原创 2019-04-01 10:06:51 · 4187 阅读 · 2 评论 -
pytorch计算两个特征的余弦相似度
首先,我们要记住一点,两个特征的余弦相似度计算出来的范围是**[-1,1]**其实,对于两个特征,它们的余弦相似度就是两个特征在经过L2归一化之后的矩阵内积。代码如下:import torchimport torch.nn.functional as F#假设feature1为N*C*W*H, feature2也为N*C*W*H(基本网络中的tensor都是这样)feature1 = ...原创 2019-07-12 15:26:27 · 23036 阅读 · 0 评论