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

DockerFile for Mac users to run DeepfaceLab with CPU Mode (#95)

Browse source 
* fix localization nullpointer exception

* fix devicelib error line:61,remove e

* support create docker from cpu dockerfile

* support preview or not when train(resolve cannot connect to X server)
This commit is contained in:
Plucky 2019-01-01 22:08:21 +08:00 • committed by iperov
parent 46bda1d683
commit a8694b73f0
11 changed files with 247 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

6
main.py
View file

@ -80,6 +80,7 @@ if __name__ == "__main__":
model_path=arguments.model_dir,
model_name=arguments.model_name,
debug = arguments.debug,
preview = arguments.preview,
#**options
batch_size = arguments.batch_size,
write_preview_history = arguments.write_preview_history,
@ -106,8 +107,9 @@ if __name__ == "__main__":
train_parser.add_argument('--force-best-gpu-idx', type=int, dest="force_best_gpu_idx", default=-1, help="Force to choose this GPU idx as best(worst).")
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('--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)
def process_convert(arguments):