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

Colab: added ability to run programs after N secs in trainer

Browse source
This commit is contained in:
iperov 2019-03-26 19:06:19 +04:00
parent 009af42617
commit fedeee4395
2 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

@ -85,6 +85,7 @@ if __name__ == "__main__":
'model_name' : arguments.model_name,
'no_preview' : arguments.no_preview,
'debug' : arguments.debug,
'execute_programs' : [ [int(x[0]), x[1] ] for x in arguments.execute_program]
}
device_args = {'cpu_only' : arguments.cpu_only,
'force_gpu_idx' : arguments.force_gpu_idx,
@ -101,6 +102,7 @@ if __name__ == "__main__":
p.add_argument('--debug', action="store_true", dest="debug", default=False, help="Debug samples.")
p.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.")
p.add_argument('--force-gpu-idx', type=int, dest="force_gpu_idx", default=-1, help="Force to choose this GPU idx.")
p.add_argument('--execute-program', dest="execute_program", action='append', nargs='+')
p.set_defaults (func=process_train)
def process_convert(arguments):