Update 12_model_ens_inference.py

This commit is contained in:
Nerdy314159265 2021-05-16 20:22:52 -05:00 committed by GitHub
parent cb076dfc80
commit 5f37ea93be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -644,7 +644,7 @@ def main():
print('done')
vr = VocalRemover(model, device, args.window_size)
vr = VocalRemover(model, device, max(args.window_size,320))
if args.tta:
pred, X_mag, X_phase = vr.inference_tta(X_spec_m, {'value': args.aggressiveness, 'split_bin': mp.param['band'][1]['crop_stop']})
@ -759,7 +759,7 @@ def main():
print('done')
vr = VocalRemover(model, device, args.window_size)
vr = VocalRemover(model, device, max(args.window_size,320))
if args.tta:
pred, X_mag, X_phase = vr.inference_tta(X_spec_m, {'value': args.aggressiveness, 'split_bin': mp.param['band'][1]['crop_stop']})
@ -874,7 +874,7 @@ def main():
print('done')
vr = VocalRemover(model, device, args.window_size)
vr = VocalRemover(model, device, max(args.window_size,320))
if args.tta:
pred, X_mag, X_phase = vr.inference_tta(X_spec_m, {'value': args.aggressiveness, 'split_bin': mp.param['band'][1]['crop_stop']})