Skip to content

Commit

Permalink
fix(pu): fix paper links in soft-argmax
Browse files Browse the repository at this point in the history
  • Loading branch information
puyuan1996 committed Dec 28, 2023
1 parent ca0a8bd commit 174370d
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions ding/torch_utils/network/soft_argmax.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,16 +6,15 @@
class SoftArgmax(nn.Module):
"""
Overview:
A neural network module that computes the SoftArgmax operation (essentially a 2-dimensional softmax),
A neural network module that computes the SoftArgmax operation (essentially a 2-dimensional spatial softmax),
which is often used for location regression tasks. It converts a feature map (such as a heatmap) into precise
coordinate locations.
Interface:
``__init__``, ``forward``
.. note::
For more information on SoftArgmax,
you can refer to the wiki page <https://wikimili.com/en/Softmax_function> or
this lecture <https://mc.ai/softmax-function-beyond-the-basics/>.
For more information on SoftArgmax, you can refer to <https://en.wikipedia.org/wiki/Softmax_function>
and the paper <https://arxiv.org/pdf/1504.00702.pdf>.
"""

def __init__(self):
Expand Down

0 comments on commit 174370d

Please sign in to comment.