search

tengshaofeng / ResidualAttentionNetwork-pytorch

a pytorch code about Residual Attention Network. This code is based on two projects from
678 stars 165 forks source link

attention map #5

Closed Xingxu1996 closed 6 years ago

Xingxu1996 commented 6 years ago

您好，请问怎么输出attention map呢

tengshaofeng commented 6 years ago

比如说Class AttentionModule_stage0中的self.softmax8_blocks就是代表某个attention map