Fix memory leak in Engine

pull/493/head
Bloodrammer 2022-02-22 17:25:54 +03:00
parent 6987d99074
commit 0e82980e8a
1 changed files with 3 additions and 3 deletions

View File

@ -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_)