Pytorch
文章平均质量分 54
alfred_torres
这个作者很懒,什么都没留下…
展开
-
cuda_knn vs torch.topk()
发现一个快速的KNN gpu算法KNN_CUDADGCNN KNN之前的knn算法用的都是DGCNN里面的torch.topk(),但是效率很低,网络训练既占用显存又慢;代替的pointnet2中的ball_query方法又不是严格的knn,可能会导致训练精度变低。### DGCNN knndef knn(x, k): inner = -2*torch.matmul(x.transpose(2, 1), x) xx = torch.sum(x**2, dim=1, keepdim=原创 2022-03-21 10:02:48 · 2379 阅读 · 4 评论 -
Pytorch源码解读-torch.utils.data
torch.utils.dataPytorch读取训练集需要用到torch.utils.data类,data类包括13个成员,主要用到的2个:class torch.utils.data.Dataset class torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, batch_s...原创 2018-09-25 11:04:08 · 2091 阅读 · 0 评论 -
Pytorch源码解读-torchvision.transforms
torchvision.transforms对于使用Pytorch的人肯定不陌生,我也用了Pytorch但是对transform却不是真正掌握,图片的预处理对网络的性能十分重要,所以打算仔细看看pytorch的源码。Transforms组成Transforms are common image transforms. They can be chained together using Co...原创 2018-11-16 20:18:09 · 1166 阅读 · 2 评论