Spaces:
Running
Running
update
Browse files
examples/clean_unet_aishell/step_2_train_model.py
CHANGED
@@ -238,6 +238,7 @@ def main():
|
|
238 |
enhanced_audios_list_r = list(enhanced_audios.detach().cpu().numpy())
|
239 |
clean_audios_list_r = list(clean_audios.detach().cpu().numpy())
|
240 |
pesq_metric = batch_pesq(enhanced_audios_list_r, clean_audios_list_r)
|
|
|
241 |
|
242 |
optimizer.zero_grad()
|
243 |
loss.backward()
|
|
|
238 |
enhanced_audios_list_r = list(enhanced_audios.detach().cpu().numpy())
|
239 |
clean_audios_list_r = list(clean_audios.detach().cpu().numpy())
|
240 |
pesq_metric = batch_pesq(enhanced_audios_list_r, clean_audios_list_r)
|
241 |
+
print(f"pesq_metric: {pesq_metric}")
|
242 |
|
243 |
optimizer.zero_grad()
|
244 |
loss.backward()
|