From 0e82980e8ab38a388af43b2cc41dc457884b53be Mon Sep 17 00:00:00 2001 From: Bloodrammer Date: Tue, 22 Feb 2022 17:25:54 +0300 Subject: [PATCH] Fix memory leak in Engine --- torchreid/engine/engine.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchreid/engine/engine.py b/torchreid/engine/engine.py index 5fe3e25..bbc01e0 100644 --- a/torchreid/engine/engine.py +++ b/torchreid/engine/engine.py @@ -367,10 +367,10 @@ class Engine(object): end = time.time() features = self.extract_features(imgs) batch_time.update(time.time() - end) - features = features.cpu().clone() + features = features.cpu() f_.append(features) - pids_.extend(pids) - camids_.extend(camids) + pids_.extend(pids.tolist()) + camids_.extend(camids.tolist()) f_ = torch.cat(f_, 0) pids_ = np.asarray(pids_) camids_ = np.asarray(camids_)