Pytorch
文章平均质量分 59
SoaringPigeon
此人不懒,但什么都不想写
展开
-
深度学习遇到梯度无法反传?看这些方法就够了!
Straight-Through Estimator (STE)Gumbel-SoftmaxReinforceConcrete / Relaxed SoftmaxContinuous RelaxationsTemperature AnnealingDifferentiable ArgMax原创 2024-03-21 12:37:32 · 233 阅读 · 2 评论 -
argparse详细使用手册
argparse使用手册转载 2022-09-16 13:22:51 · 383 阅读 · 0 评论 -
Pytorch Parallel KeyError Bug
报错:Traceback (most recent call last): File "train_point_corr.py", l原创 2022-04-08 23:50:38 · 577 阅读 · 0 评论 -
Channel Attention 通道注意力
通道注意力总结SE-Net:Squeeze-and-Excitation NetworksCBAM: Convolutional Block Attention ModuleECA-Net: Efficient Channel Attention for Deep Convolutional Neural NetworksEPSANet: An Efficient Pyramid Squeeze Attention Block on Convolutional Neural NetworkFcaNet: F原创 2021-10-21 21:15:00 · 24901 阅读 · 0 评论 -
[Repost]pytorch中的.Tensor、.tensor、.from_numpy、.as_tensor区别
转载自:pytorch中的.Tensor、.tensor、.from_numpy、.as_tensor区别转载 2021-09-01 11:45:56 · 171 阅读 · 0 评论 -
盘点表示学习(Representation Learning)发展中的逻辑
Logic in Representation Learning基于负例的对比学习:以SimCLR为例Batch之外:以MocoV2为例基于对比聚类的对比学习:负例隐身术以SwAV为例对比学习(Contrastive Learning) 是判别式自监督学习的一种,不依赖标注数据,要从无标注图像中自己学习知识。通过自动构造相似实例和不相似实例,要求习得一个表示学习模型,通过这个模型,使得相似的实例在投影空间中比较接近,而不相似的实例在投影空间中距离比较远。而如何构造相似实例,以及不相似实例,如何构造能够遵循原创 2021-07-27 01:57:27 · 537 阅读 · 0 评论 -
CUHK MMLab 3D点云目标检测框架 OpenPCDet个人学习记录
OpenPCDet个人学习记录项目地址数据处理kitti_dataset.py的理解项目地址Github地址: https://github.com/open-mmlab/OpenPCDet史少帅博士本人对OpenPCDet的描述可以参看这篇文章:OpenPCDet: Open-MMLab 面向LiDAR点云表征的3D目标检测代码库数据处理kitti_dataset.py的理解...原创 2021-06-24 22:56:47 · 1420 阅读 · 0 评论 -
查看Pytorch版本以及是否支持GPU加速
查看Pytorch版本import torchprint(torch.__version__)>>> 1.4.0 #这里便是torch版本查看Pytorch是否支持GPU加速import torchprint(torch.__version__)>>> 1.4.0 #这里便是torch版本原创 2021-04-26 20:10:13 · 3381 阅读 · 0 评论 -
torch.nn.Conv1d图文解析
nn.Conv1d 一维卷积解析对于一维卷积,我们可能以为是一个一维的卷积核在一条线上做卷积,但是这种理解是错的,一维卷积不代表卷积核只有一维,也不代表被卷积的feature也是一维。一维的意思是说卷积的方向是一维的。Example:conv1 = nn.Conv1d(in_channels=256,out_channels = 100, kernel_size = 2)input = torch.randn(32, 35, 256)# batch_size x text_len x embedd原创 2021-04-13 09:31:05 · 1531 阅读 · 0 评论 -
CUDA out of memory解决办法
CUDA out of memory解决办法当使用Pytorch GPU进行计算时经常遇到GPU存储空间过满,原因大致有两点:1.Batch_size设置过大,超过显存空间解决办法: 减小Batch_size2.之前程序运行结束后未释放显存解决办法: 按住键盘上的Win+R在弹出的框里输入cmd,进入控制台, 然后输入命令nvidia-smi查看GPU的使用情况,和占用GPU资源的程序,我们看到 PID 2492 运行着python占用很大空间可以使用如下命令杀死进程taskkill原创 2020-12-31 00:09:44 · 15089 阅读 · 3 评论