From 7d712387a183b4b71c8a1b01487a588b1ab2838b Mon Sep 17 00:00:00 2001
From: Han GuangXin <48764451+HanGuangXin@users.noreply.github.com>
Date: Wed, 27 Apr 2022 20:31:02 +0800
Subject: [PATCH] fix FP16 bug of track_id annotation

---
 yolox/core/trainer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/yolox/core/trainer.py b/yolox/core/trainer.py
index b0063399..5ab10060 100644
--- a/yolox/core/trainer.py
+++ b/yolox/core/trainer.py
@@ -91,6 +91,8 @@ def train_one_iter(self):
         iter_start_time = time.time()
 
         inps, targets = self.prefetcher.next()
+        track_ids = targets[:, :, 5]
+        targets = targets[:, :, :5]
         inps = inps.to(self.data_type)
         targets = targets.to(self.data_type)
         targets.requires_grad = False