Update train_second.py

zolero · Dec 8, 2023 · b4651a4 · b4651a4
1 parent 1ece0a3
commit b4651a4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/train_second.py b/train_second.py
@@ -11,6 +11,7 @@
 import librosa
 import click
 import shutil
+import traceback
 import warnings
 warnings.simplefilter('ignore')
 from torch.utils.tensorboard import SummaryWriter
@@ -669,7 +670,9 @@ def main(config_path):
                     loss_f += (loss_F0).mean()
 
                     iters_test += 1
-                except:
+                except Exception as e:
+                    print(f"run into exception", e)
+                    traceback.print_exc()
                     continue
 
         print('Epochs:', epoch + 1)