Merge pull request #70 from Nerdy314159265/patch-1

Alter Ensemble Scripts to Prevent Crashing
This commit is contained in:
Dilan Boskan 2021-05-17 13:14:05 +02:00 committed by GitHub
commit ae21a4195d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions

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']})

View File

@ -632,7 +632,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']})
@ -747,7 +747,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']})
@ -862,7 +862,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']})