Update ensemble_inference.py

This commit is contained in:
aufr33 2021-10-04 16:05:33 +03:00 committed by GitHub
parent c9610c4c7c
commit 01ff87da7b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -173,7 +173,7 @@ def main():
os.system('python inference.py -mt -g {} -m {} -P {} -A {} -w {} {} {} -o {} -i "{}"'.format(
args.gpu,
os.path.join('modelparams', c['model_params']) + '.json',
os.path.join('models','release','models', model_name + '.pth'),
os.path.join('models', model_name + '.pth'),
args.aggressiveness,
args.window_size,
('', '-p')[args.postprocess],