From 1fb2ac3f92742b3febc3d0420353815f756d183b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mathias=20R=C3=A9us?=
 <93187038+AwePhD@users.noreply.github.com>
Date: Wed, 18 Oct 2023 12:09:22 +0200
Subject: [PATCH] Fix  docstring for dimension of targets argument (#11058)

---
 mmdet/models/losses/triplet_loss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmdet/models/losses/triplet_loss.py b/mmdet/models/losses/triplet_loss.py
index d9c9604b8c7..4528239beb4 100644
--- a/mmdet/models/losses/triplet_loss.py
+++ b/mmdet/models/losses/triplet_loss.py
@@ -40,7 +40,7 @@ def hard_mining_triplet_loss_forward(
             inputs (torch.Tensor): feature matrix with shape
                 (batch_size, feat_dim).
             targets (torch.LongTensor): ground truth labels with shape
-                (num_classes).
+                (batch_size).
 
         Returns:
             torch.Tensor: triplet loss with hard mining.