normalize

Open Bio-shine opened this issue 5 years ago • 1 comments

# multiply & normalize ->(B, N, C') energy = torch.bmm(proj_query, proj_key) energy_new = torch.max(energy, -1, keepdim=True)[0].expand_as(energy) - energy

I think the normalization implemented in acfnet.py is different from the equation (2) in the original paper. And I wonder why you implement this way.

Oct 16 '20 03:10 Bio-shine

I have the same question. Have you figured it out?

Dec 29 '20 03:12 zzw1123