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

optimize memory usage

Browse Source
This commit is contained in:
Colombo
2020-01-04 23:51:33 +04:00
parent f95682f616
commit 2429d28737
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samplelib/SampleHost.py
+7
View File
@@ -1,5 +1,7 @@
import gc
import multiprocessing import multiprocessing
import operator import operator
import pickle
import traceback import traceback
from pathlib import Path from pathlib import Path
@@ -58,6 +60,11 @@ class SampleHost:
if result is None: if result is None:
result = SampleHost.load_face_samples( Path_utils.get_image_paths(samples_path) ) result = SampleHost.load_face_samples( Path_utils.get_image_paths(samples_path) )
result_dmp = pickle.dumps(result)
del result
gc.collect()
result = pickle.loads(result_dmp)
samples[sample_type] = result samples[sample_type] = result
elif sample_type == SampleType.FACE_TEMPORAL_SORTED: elif sample_type == SampleType.FACE_TEMPORAL_SORTED: