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

fix

Browse source
This commit is contained in:
iperov 2021-07-17 22:04:20 +04:00
parent 8b71b83c94
commit 336e0ca944
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/ModelBase.py
View file

@ -22,6 +22,7 @@ from samplelib import SampleGeneratorBase
class ModelBase(object): class ModelBase(object):
def __init__(self, is_training=False, def __init__(self, is_training=False,
is_exporting=False,
saved_models_path=None, saved_models_path=None,
training_data_src_path=None, training_data_src_path=None,
training_data_dst_path=None, training_data_dst_path=None,
@ -36,6 +37,7 @@ class ModelBase(object):
silent_start=False, silent_start=False,
**kwargs): **kwargs):
self.is_training = is_training self.is_training = is_training
self.is_exporting = is_exporting
self.saved_models_path = saved_models_path self.saved_models_path = saved_models_path
self.training_data_src_path = training_data_src_path self.training_data_src_path = training_data_src_path
self.training_data_dst_path = training_data_dst_path self.training_data_dst_path = training_data_dst_path