alibabasglab commited on
Commit
4760e88
·
verified ·
1 Parent(s): 747aa1a

Update utils/decode.py

Browse files
Files changed (1) hide show
  1. utils/decode.py +2 -2
utils/decode.py CHANGED
@@ -189,7 +189,7 @@ def decode_one_audio_frcrn_se_16k(model, device, inputs, args):
189
  outputs = model.inference(inputs).detach().cpu().numpy() # Inference on full input
190
 
191
  #normalize outputs
192
- max_abs = max(max(abs(outputs)), 1e-6))
193
  outputs = outputs / max_abs
194
  return outputs # Return the decoded audio output
195
 
@@ -444,7 +444,7 @@ def decode_one_audio_mossformer2_se_48k(model, device, inputs, args):
444
 
445
  outpus = outputs.numpy() / MAX_WAV_VALUE # Return the output normalized to [-1, 1]
446
  #normalize outputs
447
- max_abs = max(max(abs(outputs)), 1e-6))
448
  outputs = outputs / max_abs
449
 
450
  return outputs
 
189
  outputs = model.inference(inputs).detach().cpu().numpy() # Inference on full input
190
 
191
  #normalize outputs
192
+ max_abs = max(max(abs(outputs)), 1e-6)
193
  outputs = outputs / max_abs
194
  return outputs # Return the decoded audio output
195
 
 
444
 
445
  outpus = outputs.numpy() / MAX_WAV_VALUE # Return the output normalized to [-1, 1]
446
  #normalize outputs
447
+ max_abs = max(max(abs(outputs)), 1e-6)
448
  outputs = outputs / max_abs
449
 
450
  return outputs