diff --git a/run.py b/run.py index 0b61ab4..6aaa4a8 100644 --- a/run.py +++ b/run.py @@ -34,14 +34,11 @@ parser.add_argument('--gpu', help='use gpu', dest='gpu', action='store_true', de parser.add_argument('--keep-fps', help='maintain original fps', dest='keep_fps', action='store_true', default=False) parser.add_argument('--keep-frames', help='keep frames directory', dest='keep_frames', action='store_true', default=False) parser.add_argument('--max-memory', help='set max memory', default=16, type=int) -parser.add_argument('--max-cores', help='set max cpu cores', dest='cores_count', type=int) +parser.add_argument('--max-cores', help='set max cpu cores', dest='cores_count', type=int, default=psutil.cpu_count()-1) for name, value in vars(parser.parse_args()).items(): args[name] = value -if not args['cores_count']: - args['cores_count'] = psutil.cpu_count()-1 - sep = "/" if os.name == "nt": sep = "\\"