摘要:
1. cuda和cudnn版本对应关系 https://developer.nvidia.com/rdp/cudnn archive a collapse742 10 阅读全文
摘要:
torch.mul(a, b) 是矩阵 对应位相乘,即点乘操作, a和b的维度必须相等,a的维度是(1,2), 则b的维度必须是(1,2), 返回还是(1,2)的矩阵 torch.mm(a,b) 是矩阵a和b矩阵相乘,a的维度是(1,2),b的维度是(2,3),返回是(1,3)的矩阵 torch.b 阅读全文
摘要:
常用插件: Chinese(Simplified) python pylance 这是python的language server TabNine (自动补全) Bracket Pair Colorizer 2 括号 (不再维护,成为内置功能) Indent-Rainbow 让缩进带有颜色 Code 阅读全文
摘要:
1.查看数据类型 2.类型转换 方法一:简单后缀转换 方法二:使用torch.type()函数 方法三:使用type_as(tensor)将tensor转换为指定tensor的类型 3.tensor创建 指定维度和数据类型 阅读全文
摘要:
1 list 与 numpy 2 numpy 与 tensor 3 list 与 tensor 阅读全文
摘要:
1 torch.cat 将两个tensor在指定维度进行拼接 2 torch.stack 增加新的维度进行堆叠 3 torch.permute 调整tensor的维度顺序,相当于更灵活的transpose 4 tensor.contiguous view只能用在contiguous的tensor上。 阅读全文
摘要:
1. nn.Conv2d 2. nn.BatchNorm2d 3. F.avg_pool2d 阅读全文
摘要:
Drop out和BN层可以同时使用,常用的组合形式如下: CONV/FC BN ReLu Dropout CONV/FC 阅读全文
摘要:
简单记录下FCOS的创新点 1. 首次提出逐像素边框预测,对feature map上所有像素点预测。 类似于yolo v1,不过yolo v1是对7x7的cell进行预测,召回率较低,在之后v2,v3中又回归anchor based方法。 而FCOS为了提高召回率对所有像素点预测,随之而来的是很多低 阅读全文