Fix memory leak in Engine
parent
6987d99074
commit
0e82980e8a
|
@ -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_)
|
||||
|
|
Loading…
Reference in New Issue