Merge branch 'main' into main
This commit is contained in:
commit
f50337f708
5
run.py
5
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-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('--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-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():
|
for name, value in vars(parser.parse_args()).items():
|
||||||
args[name] = value
|
args[name] = value
|
||||||
|
|
||||||
if not args['cores_count']:
|
|
||||||
args['cores_count'] = psutil.cpu_count()-1
|
|
||||||
|
|
||||||
sep = "/"
|
sep = "/"
|
||||||
if os.name == "nt":
|
if os.name == "nt":
|
||||||
sep = "\\"
|
sep = "\\"
|
||||||
|
Loading…
Reference in New Issue
Block a user