目录 一、【HaloAttention】注意力机制 1.1【HaloAttention】注意力介绍 1.2【HaloAttention】核心代码 二、添加【HaloAttention】注意力机制 2.1STEP1 2.2STEP2 2.3STEP3 2.4STEP4 三、yaml文件与运行 3.1yaml文件 3.2运行成功截图 一、【HaloAttention】注意力机制 1.1【HaloAttention】注意力介绍 下图是【HaloAttention】的结构图,让我们简单分析一下运行过程和优势 处理过程: 图像分块: 输入图像大小为 4×4×𝑐,其中 𝑐 是通道数。该图像首先被分割为多个小块(如图所示被分为 4 个 2×2×𝑐的小块),每个块称为一个“block”。