2019-03-23 23:09:39 +00:00

18 lines
232 B
ReStructuredText

.. _torchreid_losses:
torchreid.losses
=================
Softmax
--------
.. automodule:: torchreid.losses.cross_entropy_loss
:members:
Triplet
-------
.. automodule:: torchreid.losses.hard_mine_triplet_loss
:members: