diff --git a/ensemble_inference.py b/ensemble_inference.py index fc6ad13..f652bc2 100644 --- a/ensemble_inference.py +++ b/ensemble_inference.py @@ -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],