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

added sumbmodules

Browse source
This commit is contained in:
leno3003 2022-03-31 10:01:10 +02:00
parent 8b4e234721
commit a874610aa7
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/ModelBase.py
View file

@ -297,6 +297,7 @@ class ModelBase(object):
def ask_target_iter(self, default_value=0):
print("Current directory")
cmd = "pwd"
print("Dentro ask_target_iter: \n\n")
os.system(cmd)
answer_filename = 'workspace/interact/interact_dict.pkl'
if os.path.exists(answer_filename):
@ -310,6 +311,9 @@ class ModelBase(object):
else:
default_target_iter = self.load_or_def_option('target_iter', default_value)
self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter))
else:
default_target_iter = self.load_or_def_option('target_iter', default_value)
self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter))
def ask_random_flip(self):
default_random_flip = self.load_or_def_option('random_flip', True)