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

Propagated BN implementation, no CLI support yet.

Browse source
This commit is contained in:
Jose 2023-02-07 13:19:45 +01:00 • committed by GitHub
parent fcd398707f
commit 7439d5003e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 1133 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

@ -132,6 +132,7 @@ if __name__ == "__main__":
'silent_start' : arguments.silent_start,
'execute_programs' : [ [int(x[0]), x[1] ] for x in arguments.execute_program ],
'debug' : arguments.debug,
'use_bn' : arguments.use_bn,
}
from mainscripts import Trainer
Trainer.main(**kwargs)
@ -149,6 +150,7 @@ if __name__ == "__main__":
p.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.")
p.add_argument('--force-gpu-idxs', dest="force_gpu_idxs", default=None, help="Force to choose GPU indexes separated by comma.")
p.add_argument('--silent-start', action="store_true", dest="silent_start", default=False, help="Silent start. Automatically chooses Best GPU and last used model.")
p.add_argument('--use-bn', action="store_true", dest="use_bn", default=False, help="Use an architecture with BatchNormalization.")
p.add_argument('--execute-program', dest="execute_program", default=[], action='append', nargs='+')
p.set_defaults (func=process_train)