github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

optimize MPSharedList.py

Browse source
This commit is contained in:
iperov 2020-12-20 23:04:00 +04:00
parent 1bd7c42b7e
commit b7bed0ef5e
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
core/mplib/MPSharedList.py
View file

@ -60,9 +60,11 @@ class MPSharedList():
break break
key -= self.obj_counts[i] key -= self.obj_counts[i]
offset_start, offset_end = struct.unpack('<QQ', bytes(sh_b[ table_offset + key*8 : table_offset + (key+2)*8]) ) sh_b = memoryview(sh_b).cast('B')
return pickle.loads( bytes(sh_b[ data_offset + offset_start : data_offset + offset_end ]) ) offset_start, offset_end = struct.unpack('<QQ', sh_b[ table_offset + key*8 : table_offset + (key+2)*8].tobytes() )
return pickle.loads( sh_b[ data_offset + offset_start : data_offset + offset_end ].tobytes() )
def __iter__(self): def __iter__(self):
for i in range(self.__len__()): for i in range(self.__len__()):