Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 3 days ago

Commit

32b6496

1 Parent(s): 049a0bc

update

Files changed (1) hide show

examples/dfnet/step_2_train_model.py CHANGED Viewed

@@ -87,7 +87,7 @@ class CollateFunction(object):
         clean_audios = torch.stack(clean_audios)
         noisy_audios = torch.stack(noisy_audios)
-        snr_db_list = torch.stack(snr_db_list)
         # assert
         if torch.any(torch.isnan(clean_audios)) or torch.any(torch.isinf(clean_audios)):

         clean_audios = torch.stack(clean_audios)
         noisy_audios = torch.stack(noisy_audios)
+        snr_db_list = torch.tensor(snr_db_list)
         # assert
         if torch.any(torch.isnan(clean_audios)) or torch.any(torch.isinf(clean_audios)):