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

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

7
samplelib/SampleHost.py
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: