Object-Detection-Knowledge-Distillation-ICLR2021 icon indicating copy to clipboard operation
Object-Detection-Knowledge-Distillation-ICLR2021 copied to clipboard

Published 20 hours ago • verified publisher icon ArchipLab-LinfengZhang

Softmax is missing in the non-local module

Open HIT-cwh opened this issue 2 years ago • 0 comments

First of all, thank you for sharing your valuable code. As shown in fig2(c) of the paper, the softmax function is used to convert the matrix product to the dot-product attention. But it is missing in the NonLocalBlockND of the released code. Moreover, it may be time-consuming to obtain this attention matrix as the matrix size is (300x300, 150x150). Looking forward to your reply. Thanks.

HIT-cwh avatar Apr 13 '22 14:04 HIT-cwh