mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #493 from Bloodrammer/master 

Fix memory leak in Engine
pull/511/head
Kaiyang 2022-02-24 00:13:56 +08:00 committed by GitHub
parent 40bfa1ae77 0e82980e8a
commit e34e3ae85f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
torchreid/engine/engine.py
View File

@ -367,10 +367,10 @@ class Engine(object):
end = time.time() end = time.time()
features = self.extract_features(imgs) features = self.extract_features(imgs)
batch_time.update(time.time() - end) batch_time.update(time.time() - end)
features = features.cpu().clone() features = features.cpu()
f_.append(features) f_.append(features)
pids_.extend(pids) pids_.extend(pids.tolist())
camids_.extend(camids) camids_.extend(camids.tolist())
f_ = torch.cat(f_, 0) f_ = torch.cat(f_, 0)
pids_ = np.asarray(pids_) pids_ = np.asarray(pids_)
camids_ = np.asarray(camids_) camids_ = np.asarray(camids_)