From 780ad5679fed940065dbb8456d8fbd7d3d8f1fea Mon Sep 17 00:00:00 2001 From: Colombo Date: Tue, 3 Mar 2020 21:25:24 +0400 Subject: [PATCH] fix --force-gpu-idxs commandline param for trainer --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 5230746..95e7960 100644 --- a/main.py +++ b/main.py @@ -176,7 +176,7 @@ if __name__ == "__main__": 'pretrained_model_path' : Path(arguments.pretrained_model_dir) if arguments.pretrained_model_dir is not None else None, 'no_preview' : arguments.no_preview, 'force_model_name' : arguments.force_model_name, - 'force_gpu_idxs' : arguments.force_gpu_idxs, + 'force_gpu_idxs' : [ int(x) for x in arguments.force_gpu_idxs.split(',') ] if arguments.force_gpu_idxs is not None else None, 'cpu_only' : arguments.cpu_only, 'execute_programs' : [ [int(x[0]), x[1] ] for x in arguments.execute_program ], 'debug' : arguments.debug,