随笔分类 - 代码相关
摘要:Recap关于transformer的基础 | 手写transformer | 理解nn.Embedding 关于transformer的基础 | 手写transformer 以notebook的形式记录下了transformer基础思路,以及手动实现transformer的少量代码。为后续reca
阅读全文
摘要:PyTorch中transformer相关类的使用 目的是大概了解一下pytorch中用transformer是什么样的逻辑。 Attention multi_head_attention_forward函数:通过一步一步看源码的调用发现其实最基本的进行attention的操作其实在这里,不过并没有
阅读全文
摘要:计算2d iou 来自OpenPCDet的eval.py。关键就在于:重叠部分的宽度 = 两个图像右边缘的较小值 - 两个图像左边缘的较大值,重叠部分的高度 = 两个图像下边缘的较小值 - 两个图像上边缘的较大值U。右小减左大,下小减上大。然后IoU等于交除以并减交 点击查看代码 @numba.ji
阅读全文