CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
core/mplib/MPSharedList.py
+4 -2
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__()):