pytorch
永远爱好技术的王师傅
这个作者很懒,什么都没留下…
展开
-
pytorch中的DataParallel
转载自知乎https://zhuanlan.zhihu.com/p/102697821原创 2020-12-22 14:00:58 · 384 阅读 · 0 评论 -
pytorch中的permute和view
总的来说,就是transpose只能用于二维矩阵的转置(可以是高维矩阵中的二维),permute可以看作是多次进行transpose,其本质是维度互换然而view必须操作在一个contiguous 的tensor上,原因是因为其操作先把tensor展成一个一维的向量,之后再reshape成一个想要的维度。因此view和permute的操作含义不一样,但是具体使用中怎么正确的使用我还没搞的特别清楚,日后填坑转载好文章https://zhuanlan.zhihu.com/p/76583143...原创 2020-12-22 10:22:17 · 430 阅读 · 0 评论 -
pytorch中tensor.topk
torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)topk的原型如上:其中k是保留的k个值,largest=True意味着选取最大的,sorted=True是指将返回结果排序topk返回的是一个tensor,第一个元素指返回的具体值,第二个元素指返回值的index直接贴代码import torchx = torch.rand(2, 3, 3).原创 2020-12-15 22:56:50 · 5373 阅读 · 1 评论 -
pytorch中BatchNorm1d、BatchNorm2d、BatchNorm3d
转载pytorch中BatchNorm1d、BatchNorm2d、BatchNorm3d - 简书 (jianshu.com)原创 2020-12-15 21:47:13 · 327 阅读 · 0 评论 -
Pytorch踩坑记录
1、使用现有模型时,如resnet、vgg16时,进行改进后,如果只想训练自己改进后的参数,可以先将原来网络参数的requires_grad = Falsenet = resnet50(pretrained = True)for param in net.parameters(): param.requires_grad = False#添加自己定义的层net.fc = n...原创 2020-04-13 00:19:29 · 635 阅读 · 1 评论