StarMirror/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2024-03-22 13:10:55 +08:00
Code Issues Projects Releases Wiki Activity

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
Show Stats 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):
def __init__(self, is_training=False,
is_exporting=False,
saved_models_path=None,
training_data_src_path=None,
training_data_dst_path=None,
@ -36,6 +37,7 @@ class ModelBase(object):
silent_start=False,
**kwargs):
self.is_training = is_training
self.is_exporting = is_exporting
self.saved_models_path = saved_models_path
self.training_data_src_path = training_data_src_path
self.training_data_dst_path = training_data_dst_path