yuyongcan / STAMP

The official repository of ECCV 2024 paper "Outlier-Aware Test-time Adaptation with Stable Memory Replay"
13 stars 0 forks source link

Double softmax in OOD score computation of stamp #2

Closed MarcLafon closed 2 months ago

MarcLafon commented 3 months ago

It seems that the ood score used in stamp is obtained via a double softmax operation. The outupt of the forward function of stamp is already a probability vector as it is computed with the softmax operator (lines 184-189 of the stamp.py file). Then the ood score in the get_accuracy function in the utils.py file is again computed using the softmax_entropy function (line 256).

yuyongcan commented 2 months ago

Thanks for your correction. We fixed the bug in our latest version.