github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-20 21:43:22 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
iperov 2021-10-01 19:24:22 +04:00
commit 03212982e6
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
xlib/avecl/_internal/backend/Device.py
View file

@ -210,10 +210,12 @@ class Device:
"""
pool = self._pooled_buffers
mems = [ (k,x) for k in pool.keys() for x in pool[k] ]
mems = random.sample(mems, max(1,int(len(mems)*0.25)) )
for k, mem in mems:
self._cl_mem_free(mem)
pool[k].remove(mem)
if len(mems) != 0:
mems = random.sample(mems, max(1,int(len(mems)*0.25)) )
for k, mem in mems:
self._cl_mem_free(mem)
pool[k].remove(mem)
return len(mems) != 0
def _keep_target_memory_usage(self):