mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-07 05:22:06 -07:00
Update main.py
This commit is contained in:
parent
64537204bb
commit
cd9e2c239c
1 changed files with 0 additions and 2 deletions
2
main.py
2
main.py
|
@ -80,7 +80,6 @@ if __name__ == "__main__":
|
||||||
model_path=arguments.model_dir,
|
model_path=arguments.model_dir,
|
||||||
model_name=arguments.model_name,
|
model_name=arguments.model_name,
|
||||||
debug = arguments.debug,
|
debug = arguments.debug,
|
||||||
preview = arguments.preview,
|
|
||||||
#**options
|
#**options
|
||||||
batch_size = arguments.batch_size,
|
batch_size = arguments.batch_size,
|
||||||
write_preview_history = arguments.write_preview_history,
|
write_preview_history = arguments.write_preview_history,
|
||||||
|
@ -108,7 +107,6 @@ if __name__ == "__main__":
|
||||||
train_parser.add_argument('--multi-gpu', action="store_true", dest="multi_gpu", default=False, help="MultiGPU option. It will select only same best(worst) GPU models.")
|
train_parser.add_argument('--multi-gpu', action="store_true", dest="multi_gpu", default=False, help="MultiGPU option. It will select only same best(worst) GPU models.")
|
||||||
train_parser.add_argument('--force-gpu-idxs', type=str, dest="force_gpu_idxs", default=None, help="Override final GPU idxs. Example: 0,1,2.")
|
train_parser.add_argument('--force-gpu-idxs', type=str, dest="force_gpu_idxs", default=None, help="Override final GPU idxs. Example: 0,1,2.")
|
||||||
train_parser.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.")
|
train_parser.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.")
|
||||||
train_parser.add_argument('--preview', action="store_true",dest="preview", default=False, help="Show preview.")
|
|
||||||
|
|
||||||
train_parser.set_defaults (func=process_train)
|
train_parser.set_defaults (func=process_train)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue