添加CrissCrossAttention注意力机制
1. CrissCrossAttention注意力机制论文
论文题目:CCNet: Criss-Cross Attention for Semantic
Segmentation
论文链接:CCNet: Criss-Cross Attention for Semantic
Segmentation
2. CrissCrossAttention注意力机制原理
3. CrissCrossAttention注意力机制的配置
3.1common.py配置
./models/common.py文件增加以下模块
在最后添加如下代码块:
from torch.nn import Softmax